1from UM.Scene.Iterator.DepthFirstIterator import DepthFirstIterator 2from UM.Scene.SceneNode import SceneNode 3 4 5def test_DepthFirstIterator(): 6 7 root_node = SceneNode() 8 9 node_1 = SceneNode() 10 root_node.addChild(node_1) 11 12 node_1_child_1 = SceneNode() 13 node_1_child_2 = SceneNode() 14 15 node_1.addChild(node_1_child_1) 16 node_1.addChild(node_1_child_2) 17 18 node_2 = SceneNode() 19 root_node.addChild(node_2) 20 21 node_2_child_1 = SceneNode() 22 node_2.addChild(node_2_child_1) 23 24 assert list(DepthFirstIterator(root_node)) == [root_node, node_1, node_2, node_1_child_1, node_1_child_2, node_2_child_1]