test_network.py 文件源码

def test_rev(self):
        """Test if taking turning the graphml back into networkx yields the expected edges and nodes"""
        rev = nx.read_graphml(self.path)
        self.assertSetEqual({'Alice', 'Bobby', 'file01', 'file02', 'file02', 'file03'}, set(rev.nodes()))
        self.assertSetEqual({'file01', 'file02'}, set(rev.edge['Alice']))
        self.assertSetEqual({'file01', 'file03'}, set(rev.edge['Bobby']))
        self.assertSetEqual({'Alice', 'Bobby'}, set(rev.edge['file01']))
        self.assertSetEqual({'Alice'}, set(rev.edge['file02']))
        self.assertSetEqual({'Bobby'}, set(rev.edge['file03']))