test_network.py 文件源码

def test_rev_edge_attr(self):
        """Are edge attr preserved when we convert the graphml back to a network?"""
        rev = nx.read_graphml(self.path)
        mg = self.mg
        # Check that edge attributes still exist
        for n1, n2, data in self.mg.edges(data=True):
            self.assertIn('sha', rev.edge[n1][n2])

        # Check that edge attributes are correct (atomic and vector)
        for n1, n2, data in mg.edges(data=True):
            for k in data:
                if not isinstance(data[k], list):
                    self.assertEqual(data[k], rev.edge[n1][n2][k])
                else:
                    self.assertEqual(gitnet.helpers.list_to_scd(data[k]), rev.edge[n1][n2][k])