Add BBox to Geometries.

Author: eseglem

geojson_pydantic/geometries.py

@@ -1,14 +1,15 @@
 """pydantic models for GeoJSON Geometry objects."""
 
 import abc
-from typing import Any, Iterator, List, Literal, Union
+from typing import Any, Iterator, List, Literal, Optional, Union
 
 from pydantic import BaseModel, Field, ValidationError, validator
 from pydantic.error_wrappers import ErrorWrapper
 from typing_extensions import Annotated
 
 from geojson_pydantic.geo_interface import GeoInterfaceMixin
 from geojson_pydantic.types import (
+    BBox,
     LinearRing,
     LineStringCoords,
     MultiLineStringCoords,
@@ -55,6 +56,7 @@ class _GeometryBase(BaseModel, abc.ABC, GeoInterfaceMixin):
 
     type: str
     coordinates: Any
+    bbox: Optional[BBox] = None
 
     @property
     @abc.abstractmethod
@@ -238,6 +240,7 @@ class GeometryCollection(BaseModel, GeoInterfaceMixin):
 
     type: Literal["GeometryCollection"]
     geometries: List[Geometry]
+    bbox: Optional[BBox] = None
 
     def __iter__(self) -> Iterator[Geometry]:  # type: ignore [override]
         """iterate over geometries"""

tests/test_features.py

@@ -168,7 +168,7 @@ class Pointy:
         __geo_interface__ = {"type": "Point", "coordinates": (0.0, 0.0)}
 
     feat = Feature(type="Feature", geometry=Pointy(), properties={})
-    assert feat.geometry.dict() == Pointy.__geo_interface__
+    assert feat.geometry.dict(exclude_unset=True) == Pointy.__geo_interface__
 
 
 def test_feature_with_null_geometry():