def test_bipartite(self): mapping = network_flow.calculate_entity_mapping(self.G, method='bipartite') self.assertDictEqual(self.expected_map, mapping)
def test_weighted(self): mapping = network_flow.calculate_entity_mapping(self.G, method='weighted') self.assertDictEqual(self.expected_map, mapping)
def test_default(self): mapping = network_flow.calculate_entity_mapping(self.G) self.assertDictEqual(self.expected_map, mapping)