Merge pull request #136 from developmentseed/resolveCircular

resolve circular deps between geometry and geometryCollection

Author: vincentsarago

geojson_pydantic/geometries.py

@@ -319,6 +319,8 @@ def check_geometries(cls, geometries: List) -> List:
     Field(discriminator="type"),
 ]
 
+GeometryCollection.update_forward_refs()
+
 
 def parse_geometry_obj(obj: Any) -> Geometry:
     """

tests/test_features.py

@@ -71,6 +71,18 @@ class GenericProperties(BaseModel):
 }
 
 
+@pytest.mark.parametrize(
+    "obj",
+    [
+        FeatureCollection,
+        Feature,
+    ],
+)
+def test_pydantic_schema(obj):
+    """Test schema for Pydantic Object."""
+    assert obj.model_json_schema()
+
+
 def test_feature_collection_iteration():
     """test if feature collection is iterable"""
     gc = FeatureCollection(

tests/test_geometries.py

@@ -18,6 +18,23 @@
 )
 
 
+@pytest.mark.parametrize(
+    "obj",
+    [
+        GeometryCollection,
+        LineString,
+        MultiLineString,
+        MultiPoint,
+        MultiPolygon,
+        Point,
+        Polygon,
+    ],
+)
+def test_pydantic_schema(obj):
+    """Test schema for Pydantic Object."""
+    assert obj.model_json_schema()
+
+
 def assert_wkt_equivalence(geom: Union[Geometry, GeometryCollection]):
     """Assert WKT equivalence with Shapely."""
     # Remove any trailing `.0` to match Shapely format