test: Update yaml tree test model

If any child of a model is not an ASDF pydantic model, the YAML tree will be the return of pydantic's model dump (a dict of native objects).
ketozhang · Jun 21, 2024 · 4f8b06a · 4f8b06a
1 parent 09268b6
commit 4f8b06a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/asdf_pydantic/examples/tree.py b/asdf_pydantic/examples/tree.py
@@ -14,4 +14,4 @@ class Node(BaseModel):
 class AsdfNode(Node, AsdfPydanticModel):
     _tag = "asdf://asdf-pydantic/examples/tags/node-1.0.0"
 
-    child: Optional[Union[Node, AsdfNode]] = None
+    child: Optional[Union[AsdfNode, Node]] = None
diff --git a/tests/convert_to_asdf_yaml_tree_test.py b/tests/convert_to_asdf_yaml_tree_test.py
@@ -21,5 +21,5 @@ def test_given_mix_child_is_mix_of_AsdfNode_and_Node():
     }
 
     assert AsdfNode(child=Node(child=AsdfNode())).asdf_yaml_tree() == {
-        "child": {"child": AsdfNode()}
+        "child": {"child": {"child": None}}
     }