code: def test_get_nodes_at_depth():
      root = build_full_tree(5)
  
      def test_for_node(node):
          assert node.get_nodes_at_depth(0) == [node]
          assert node.get_nodes_at_depth(1) == node.children
          assert (
              node.get_nodes_at_depth(2)
              == node.children[0].children + node.children[1].children
          )
  
      test_for_node(root)
      test_for_node(root.children[0])
      test_for_node(root.children[1])
  
imports: from orion.core.utils.tree import DepthFirstTraversal, PreOrderTraversal, TreeNode, flattened
label_desc: Use image node_id 26 for calling a global function with example usage: test_get_nodes_at_depth() without return types