def test_od_pairs_from_topology_undirected(self): topology = fnss.ring_topology(3) topology.add_path([7, 8, 9]) # isolated node: no flows from/to this node od_pairs = fnss.od_pairs_from_topology(topology) expected_od_pairs = [(0, 1), (0, 2), (1, 0), (1, 2), (2, 0), (2, 1), (7, 8), (7, 9), (8, 7), (8, 9), (9, 7), (9, 8)] self.assertEquals(len(expected_od_pairs), len(od_pairs)) for od in expected_od_pairs: self.assertTrue(od in od_pairs)
def test_od_pairs_from_topology_directed(self): dir_topology = fnss.DirectedTopology() dir_topology.add_edge(0, 1) dir_topology.add_edge(1, 0) dir_topology.add_edge(1, 2) dir_topology.add_edge(3, 2) dir_topology.add_edge(8, 9) expected_od_pairs = [(0, 1), (0, 2), (1, 0), (1, 2), (3, 2), (8, 9)] od_pairs = fnss.od_pairs_from_topology(dir_topology) self.assertEquals(len(expected_od_pairs), len(od_pairs)) for od in expected_od_pairs: self.assertTrue(od in od_pairs)
def test_od_pairs_from_topology_directed(self): dir_topology = fnss.DirectedTopology() dir_topology.add_edge(0, 1) dir_topology.add_edge(1, 0) dir_topology.add_edge(1, 2) dir_topology.add_edge(3, 2) dir_topology.add_edge(8, 9) expected_od_pairs = [(0, 1), (0, 2), (1, 0), (1, 2), (3, 2), (8, 9)] od_pairs = fnss.od_pairs_from_topology(dir_topology) self.assertEqual(len(expected_od_pairs), len(od_pairs)) for od in expected_od_pairs: self.assertTrue(od in od_pairs)