Test data fixed!

References #11

Author: exhuma

test.py

@@ -71,22 +71,22 @@ class HClusterIntegerTestCase(unittest.TestCase):
 
     def setUp(self):
         self.__data = [791, 956, 676, 124, 564, 84, 24, 365, 594, 940, 398,
-                            971, 131, 365, 542, 336, 518, 835, 134, 391]
+                       971, 131, 365, 542, 336, 518, 835, 134, 391]
 
     def testCluster(self):
         "Basic Hierarchical Clustering test with integers"
         cl = HierarchicalClustering(self.__data, lambda x, y: abs(x - y))
         cl.cluster()
         self.assertEqual([
-                [24],
-                [84, 124, 131, 134],
-                [336, 365, 365, 365, 398, 391],
-                [940, 956, 971],
-                [791],
-                [835],
-                [676],
-                [518, 564, 542]],
-                cl.getlevel(40))
+            [24],
+            [84, 124, 131, 134],
+            [336, 365, 365, 391, 398],
+            [676],
+            [594, 518, 542, 564],
+            [940, 956, 971],
+            [791],
+            [835],
+        ], cl.getlevel(40))
 
     def testUnmodifiedData(self):
         cl = HierarchicalClustering(self.__data, lambda x, y: abs(x - y))
@@ -102,9 +102,9 @@ def sim(self, x, y):
         return 1 - sm.ratio()
 
     def setUp(self):
-        self.__data = ("Lorem ipsum dolor sit amet, consectetuer adipiscing "
-                "elit. Ut elit. Phasellus consequat ultricies mi. Sed congue "
-                "leo at neque. Nullam."). split()
+        self.__data = ("Lorem ipsum dolor sit amet consectetuer adipiscing "
+                       "elit Ut elit Phasellus consequat ultricies mi Sed "
+                       "congue leo at neque Nullam").split()
 
     def testDataTypes(self):
         "Test for bug #?"
@@ -118,19 +118,17 @@ def testCluster(self):
         "Basic Hierachical clustering test with strings"
         cl = HierarchicalClustering(self.__data, self.sim)
         self.assertEqual([
-                    ['Nullam.'],
-                    ['Sed'],
-                    ['mi.'],
-                    ['ultricies'],
-                    ['Phasellus'],
-                    ['amet,', 'at'],
-                    ['sit', 'elit.', 'elit.', 'elit.'],
-                    ['leo', 'Lorem', 'dolor'],
-                    ['neque.', 'congue', 'consectetuer', 'consequat'],
-                    ['ipsum'],
-                    ['adipiscing']
-                    ],
-                cl.getlevel(0.5))
+            ['ultricies'],
+            ['Sed'],
+            ['Phasellus'],
+            ['mi'],
+            ['Nullam'],
+            ['sit', 'elit', 'elit', 'Ut', 'amet', 'at'],
+            ['leo', 'Lorem', 'dolor'],
+            ['congue', 'neque', 'consectetuer', 'consequat'],
+            ['adipiscing'],
+            ['ipsum'],
+        ], cl.getlevel(0.5))
 
     def testUnmodifiedData(self):
         cl = HierarchicalClustering(self.__data, self.sim)