GitHub Access Token became invalid

It seems like the GitHub access token used for retrieving details about this repository from GitHub became invalid. This might prevent certain types of inspections from being run (in particular, everything related to pull requests).
Please ask an admin of your repository to re-new the access token on this website.
Passed
Push — master ( 7c1057...9930c8 )
by Andreas
01:11
created

klib.tests.test_preprocess   A

Complexity

Total Complexity 8

Size/Duplication

Total Lines 56
Duplicated Lines 0 %

Importance

Changes 0
Metric Value
eloc 46
dl 0
loc 56
rs 10
c 0
b 0
f 0
wmc 8

2 Methods

Rating   Name   Duplication   Size   Complexity  
A Test_train_dev_test_split.setUpClass() 0 14 1
B Test_train_dev_test_split.test_train_dev_test_split() 0 30 7
1
import pandas as pd
2
import unittest
3
from klib.preprocess import train_dev_test_split
4
5
if __name__ == '__main__':
6
    unittest.main()
7
8
9
class Test_train_dev_test_split(unittest.TestCase):
10
11
    @classmethod
12
    def setUpClass(cls):
13
        cls.data_split = pd.DataFrame([[1, 2, 3, 4, 'a'],
14
                                       [2, 4, 5, 6, 'b'],
15
                                       [3, 4, 2, 1, 'c'],
16
                                       [4, 0, 3, 4, 'a'],
17
                                       [5, 4, 5, 6, 'b'],
18
                                       [6, 4, 2, 1, 'c'],
19
                                       [7, 0, 3, 4, 'a'],
20
                                       [8, 4, 5, 6, 'b'],
21
                                       [9, 4, 2, 1, 'c'],
22
                                       [10, 2, 1, 5, 'b']],
23
                                      columns=['Col1', 'Col2', 'Col3', 'Col4', 'Col5'])
24
        cls.data_target = pd.Series([1, 0, 1, 0, 0, 1, 1, 0, 1, 1])
25
26
    def test_train_dev_test_split(self):
27
        # Test the propper splitting in train, dev and test sets
28
        expected_results = [(8, 4), (1, 4), (1, 4), (8,), (1,), (1,)]
29
        for i, _ in enumerate(expected_results):
30
            self.assertEqual(train_dev_test_split(self.data_split, 'Col2')[i].shape, expected_results[i])
31
32
        expected_results = [(6, 5), (2, 5), (2, 5), (6,), (2,), (2,)]
33
        for i, _ in enumerate(expected_results):
34
            self.assertEqual(train_dev_test_split(self.data_split, target=self.data_target,
35
                                                  dev_size=0.2, test_size=0.2)[i].shape, expected_results[i])
36
37
        expected_results = [(8, 5), (2, 5), (8,), (2,)]
38
        for i, _ in enumerate(expected_results):
39
            self.assertEqual(train_dev_test_split(self.data_split, target=self.data_target,
40
                                                  dev_size=0, test_size=0.2)[i].shape, expected_results[i])
41
42
        expected_results = [(8, 4), (2, 4), (8,), (2,)]
43
        for i, _ in enumerate(expected_results):
44
            self.assertEqual(train_dev_test_split(self.data_split, target='Col2',
45
                                                  dev_size=0, test_size=0.2)[i].shape, expected_results[i])
46
47
        expected_results = [(5, 5), (5, 5), (5,), (5,)]
48
        for i, _ in enumerate(expected_results):
49
            self.assertEqual(train_dev_test_split(self.data_split, target=self.data_target,
50
                                                  dev_size=0.5, test_size=0)[i].shape, expected_results[i])
51
52
        expected_results = [(5, 4), (5, 4), (5,), (5,)]
53
        for i, _ in enumerate(expected_results):
54
            self.assertEqual(train_dev_test_split(self.data_split, target='Col2',
55
                                                  dev_size=0.5, test_size=0)[i].shape, expected_results[i])
56