def test_two_go(self):
edges = [(1, 2), (2, 3), (3, 4)]
G = nx.DiGraph()
G.add_edges_from(edges)
result = clean_graph(G)
gold = [1, 4]
self.assertItemsEqual(result, gold)
def test_complex(self):
G = nx.DiGraph()
G.add_edges_from([(1, 2), (2, 3)])
gold = nx.DiGraph()
gold.add_edges_from([(1, 3)])
clean = clean_graph(G)
self.assertEqual(clean.edges(), gold.edges())
def test_one_goes(self):
edges = [(1, 2), (2, 3)]
G = nx.DiGraph()
G.add_edges_from(edges)
result = clean_graph(G)
gold = [1, 3]
self.assertItemsEqual(result, gold)
def test_direct(self):
edges = [(1, 2),]
G = nx.DiGraph()
G.add_edges_from(edges)
result = clean_graph(G)
gold = [1, 2]
self.assertItemsEqual(result, gold)
def test_empty(self):
edges = []
G = nx.DiGraph()
G.add_edges_from(edges)
result = clean_graph(G)
gold = []
self.assertItemsEqual(result, gold)
def test_one(self):
G = nx.DiGraph()
G.add_node(1)
clean = clean_graph(G)
gold = {1}
self.assertItemsEqual(clean.nodes(), gold)
def test_empty(self):
G = nx.DiGraph()
gold = nx.DiGraph()
self.assertGraphEqual(clean_graph(G), gold)
self.assertEqual(G.edges(), gold.edges())