Merge pull request #20 from zStupan/fix-tests

fix dataset + reformat test files

Author: firefly-cpp

niaarm/dataset.py

@@ -41,7 +41,7 @@ def __analyse_types(self):
                 unique_categories = None
             else:
                 dtype = "cat"
-                unique_categories = sorted(col.astype('string').unique().tolist(), key=str.lower)
+                unique_categories = sorted(col.astype(str).unique().tolist(), key=str.lower)
                 min_value = None
                 max_value = None
 

niaarm/tests/test_coverage.py

@@ -3,6 +3,7 @@
 from niaarm.dataset import Dataset
 import os
 
+
 class TestCoverage(TestCase):
     # let's borrow a test case from wikipedia:
     # https://en.wikipedia.org/wiki/Lift_(data_mining)

niaarm/tests/test_cut_point.py

@@ -3,6 +3,7 @@
 from niaarm.dataset import Dataset
 import os
 
+
 class TestCutPoint(TestCase):
     # let's borrow a test case from Wikipedia:
     # https://en.wikipedia.org/wiki/Lift_(data_mining)

niaarm/tests/test_read_csv.py

@@ -2,6 +2,7 @@
 from niaarm.dataset import Dataset
 import os
 
+
 class TestReadCSVAbalone(TestCase):
     def test_read_features(self):
 

niaarm/tests/test_rule_building.py

@@ -3,6 +3,7 @@
 from niaarm.dataset import Dataset
 import os
 
+
 class TestBuildRuleA(TestCase):
     # let's borrow a test case from Wikipedia:
     # https://en.wikipedia.org/wiki/Lift_(data_mining)

niaarm/tests/test_shrinkage.py

@@ -3,6 +3,7 @@
 from niaarm.dataset import Dataset
 import os
 
+
 class TestShrinkageA(TestCase):
     def setUp(self):
         data = Dataset(os.path.join(os.path.dirname(__file__), 'test_data', 'wiki_test_case.csv'))

niaarm/tests/test_support_confidence.py

@@ -3,6 +3,7 @@
 from niaarm.dataset import Dataset
 import os
 
+
 class TestSupportConfidence(TestCase):
     # let's borrow test case from wikipedia:
     # https://en.wikipedia.org/wiki/Lift_(data_mining)