Added max k clustering

prakhar1989 · prakhar1989 · commit 7c7c8769519e · 2012-12-14T13:40:24.000+05:30
diff --git a/tests/assign.py b/tests/assign.py
diff --git a/tests/digraph_test.py b/tests/digraph_test.py
@@ -42,9 +42,6 @@ def test_del_nonexistent_node_throws_exception(self):
     def test_add_duplicate_edge_throws_exception(self):
         self.assertRaises(Exception, self.gr.add_edge, ("a", "b"))
 
-    def test_add_edge_from_non_existing_node(self):
-        self.assertRaises(Exception, self.gr.add_edge, ("b", "z"))
-
     def test_adding_self_loop(self):
         self.gr.add_edge(("a", "a"))
         self.assertTrue(self.gr.has_edge(("a", "a")))
@@ -77,25 +74,6 @@ def test_set_weight_on_existing_edge(self):
     def test_weight_for_nonexisting_edge(self):
         self.assertRaises(Exception, self.gr.get_edge_weight, ("a", "c"))
 
-    def test_set_edge_properties(self):
-        self.gr.set_edge_properties(('a', 'b'), label="label")
-        self.assertEqual(self.gr.get_edge_properties(('a', 'b'))["label"], "label")
-
-    def test_add_node_properties_of_existing_node(self):
-        self.gr.add_node_attribute("a", "expensive")
-        self.assertEqual(self.gr.node_attributes("a"), "expensive")
-
-    def test_get_node_properties_of_blank_node(self):
-        self.assertEqual(self.gr.node_attributes("a"), [])
-
-    def test_add_node_attribute_of_non_existing_node(self):
-        self.assertRaises(Exception, self.gr.add_node_attribute, "z")
-
-    def test_clear_node_attributes_method(self):
-        self.gr.add_node_attribute("a", "expensive")
-        self.gr.clear_node_attributes()
-        self.assertEqual(self.gr.node_attributes("a"), [])
-
     def test_get_transpose_method(self):
         transpose = self.gr.get_transpose()
         self.assertEqual(len(transpose.nodes()), len(self.gr.nodes()))
diff --git a/tests/graph_algorithms_test.py b/tests/graph_algorithms_test.py
@@ -126,7 +126,6 @@ def test_kruskals_minimum_spanning_tree(self):
             if v not in gr.nodes():
                 gr.add_node(v)
             gr.add_edge( (u, v), int(w) )
-
         min_cost = kruskal_MST(gr)
         self.assertEqual(min_cost, 39)
 
diff --git a/tests/graph_test.py b/tests/graph_test.py
@@ -88,24 +88,5 @@ def test_set_weight_on_existing_edge(self):
     def test_weight_for_nonexisting_edge(self):
         self.assertRaises(Exception, self.gr.get_edge_weight, ("a", "c"))
 
-    def test_set_edge_properties(self):
-        self.gr.set_edge_properties(('a', 'b'), label="label")
-        self.assertEqual(self.gr.get_edge_properties(('a', 'b'))["label"], "label")
-
-    def test_add_node_properties_of_existing_node(self):
-        self.gr.add_node_attribute("a", "expensive")
-        self.assertEqual(self.gr.node_attributes("a"), "expensive")
-
-    def test_get_node_properties_of_blank_node(self):
-        self.assertEqual(self.gr.node_attributes("a"), [])
-
-    def test_add_node_attribute_of_non_existing_node(self):
-        self.assertRaises(Exception, self.gr.add_node_attribute, "z")
-
-    def test_clear_node_attributes_method(self):
-        self.gr.add_node_attribute("a", "expensive")
-        self.gr.clear_node_attributes()
-        self.assertEqual(self.gr.node_attributes("a"), [])
-
 if __name__ == "__main__":
     unittest.main()
diff --git a/tests/unionfind_test.py b/tests/unionfind_test.py
@@ -24,12 +24,21 @@ def test_insert_method(self):
         self.assertEqual(self.uf.get_leader("c"), self.uf.get_leader("d"))
         self.assertEqual(self.uf.get_leader("a"), self.uf.get_leader("d"))
 
+    def test_insert_one_node(self):
+        self.uf.insert('z')
+        self.assertEqual(self.uf.get_leader('z'), 'z')
+        self.assertEqual(self.uf.count_groups(), 3)
+
     def test_make_union_method(self):
         self.uf.make_union(self.uf.get_leader("a"), self.uf.get_leader("i"))
         self.assertEqual(self.uf.get_leader("a"), self.uf.get_leader("i"))
 
     def test_make_union_with_invalid_leader_raises_exception(self):
         self.assertRaises(Exception, self.uf.make_union, "a", "z")
 
+    def test_get_count(self):
+        self.uf.insert("z", "y")
+        self.assertEqual(self.uf.count_groups(), 3)
+
 if __name__ == "__main__":
     unittest.main()
