Commit ce265d6d authored by Kelvinrr's avatar Kelvinrr
Browse files

fixed mst node comparison issue 

changed subgraph from edges and added sort compare
parent 4540712b

autocnet/graph/tests/test_network.py

+4 −5
Original line number Diff line number Diff line
@@ -131,12 +131,11 @@ class TestGraphMasks(unittest.TestCase): 


        cls.graph = network.CandidateGraph.from_adjacency(cls.test_dict)

        cls.graph.minimum_spanning_tree()

        cls.mst_graph = cls.graph.copy()

        removed_edges = cls.graph.graph_masks['mst'][cls.graph.graph_masks['mst'] == False].index



        for s, d, edge in cls.graph.edges_iter(data=True):

            if not cls.graph.graph_masks['mst'][(s, d)]:

                cls.mst_graph.remove_edge(s, d)

        cls.mst_graph = cls.graph.copy()

        cls.mst_graph.remove_edges_from(removed_edges)



    def test_mst_output(self):

        self.assertEqual(self.mst_graph.nodes(), self.graph.nodes())

        self.assertEqual(sorted(self.mst_graph.nodes()), sorted(self.graph.nodes()))

        self.assertEqual(self.mst_graph.number_of_edges(), self.graph.number_of_edges()-5)
 
 No newline at end of file