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_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())