def multi_levels_test(self):
deps = DepsGraph()
deps.add_node(1)
deps.add_node(2)
deps.add_node(32)
deps.add_node(31)
deps.add_edge(1, 2)
deps.add_edge(2, 31)
deps.add_edge(2, 32)
self.assertEqual([[31, 32], [2], [1]], deps.by_levels())
def basic_levels_test(self):
deps = DepsGraph()
deps.add_node(1)
deps.add_node(2)
deps.add_node(3)
deps.add_edge(1, 2)
deps.add_edge(2, 3)
self.assertEqual([[3], [2], [1]], deps.by_levels())
def basic_levels_test(self):
deps = DepsGraph()
deps.add_node(1)
deps.add_node(2)
deps.add_node(3)
deps.add_edge(1, 2)
deps.add_edge(2, 3)
self.assertEqual([[3], [2], [1]], deps.by_levels())
def multi_levels_test3(self):
deps = DepsGraph()
deps.add_node(1)
deps.add_node(5)
deps.add_node(2)
deps.add_node(32)
deps.add_node(31)
deps.add_edge(1, 2)
deps.add_edge(1, 5)
deps.add_edge(2, 31)
deps.add_edge(2, 32)
deps.add_edge(32, 5)
self.assertEqual([[5, 31], [32], [2], [1]], deps.by_levels())
