def test_one(self): self.G.add_edges_from([(1, 2)]) root = _get_root(self.G) gold = 1 self.assertEqual(root, gold)
def test_two(self): self.G.add_edges_from([(1, 2), (2, 3), (3, 4), (4, 6), (3, 5), (4, 7)]) root = _get_root(self.G) gold = 1 self.assertEqual(root, gold)