def test_node_init_with_inputs(): with wrap() as wrapper: node = Node("foo.txt", wrapper, [Node("bar.txt", wrapper)]) assert node.key == "foo.txt" assert node.inputs[0].key == "bar.txt" expected = {0: "bar.txt", 1: "foo.txt"} for i, n in enumerate(node.walk_inputs()): assert expected[i] == n.key