feat!: migrate example models to tests/ and remove example extension …

…from package (#39)
ketozhang · Nov 17, 2024 · 28a01b5 · 28a01b5
1 parent 5cbeb18
commit 28a01b5
Show file tree

Hide file tree

Showing 9 changed files with 5 additions and 43 deletions.
diff --git a/asdf_pydantic/examples/__init__.py b/asdf_pydantic/examples/__init__.py
diff --git a/asdf_pydantic/examples/extensions.py b/asdf_pydantic/examples/extensions.py
diff --git a/pyproject.toml b/pyproject.toml
@@ -34,9 +34,6 @@ Documentation = "https://asdf-pydantic.readthedocs.io"
 Issues = "https://github.com/ketozhang/asdf-pydantic/issues"
 Source = "https://github.com/ketozhang/asdf-pydantic"
 
-[project.entry-points]
-'asdf.extensions' = { asdf_pydantic_extension = 'asdf_pydantic.examples.extensions:get_extensions' }
-
 [tool.hatch.version]
 source = "vcs"
 

diff --git a/tests/convert_to_asdf_yaml_tree_test.py b/tests/convert_to_asdf_yaml_tree_test.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from asdf_pydantic.examples.tree import AsdfTreeNode, Node
+from tests.examples.tree import AsdfTreeNode, Node
 
 
 def test_sanity():

diff --git a/tests/entry_point_test.py b/tests/entry_point_test.py
diff --git a/asdf_pydantic/examples/shapes.py → tests/examples/shapes.py b/asdf_pydantic/examples/shapes.py → tests/examples/shapes.py
diff --git a/tests/examples/test_rectangle.py b/tests/examples/test_rectangle.py
@@ -7,13 +7,8 @@
 import yaml
 from asdf.extension import Extension
 
-from asdf_pydantic import AsdfPydanticConverter, AsdfPydanticModel
-
-
-class AsdfRectangle(AsdfPydanticModel):
-    _tag = "asdf://asdf-pydantic/examples/tags/rectangle-1.0.0"
-    width: float
-    height: float
+from asdf_pydantic import AsdfPydanticConverter
+from tests.examples.shapes import AsdfRectangle
 
 
 @pytest.fixture()

diff --git a/asdf_pydantic/examples/tree.py → tests/examples/tree.py b/asdf_pydantic/examples/tree.py → tests/examples/tree.py
diff --git a/tests/schema_validation_test.py b/tests/schema_validation_test.py
@@ -7,8 +7,8 @@
 from asdf.extension import Extension
 
 from asdf_pydantic import AsdfPydanticConverter
-from asdf_pydantic.examples.shapes import AsdfRectangle
-from asdf_pydantic.examples.tree import AsdfTreeNode
+from tests.examples.shapes import AsdfRectangle
+from tests.examples.tree import AsdfTreeNode
 
 
 def setup_module():