Switch WKT to use type over class name.

CHANGELOG.md

@@ -7,6 +7,10 @@ and this project adheres to [Semantic Versioning](http://semver.org/).
 
 ## [0.5.0] - TBD
 
+## Fixed
+
+- Derive WKT type from Geometry's type instead of class name (author @eseglem, https://github.com/developmentseed/geojson-pydantic/pull/81)
+
 ### Changed
 
 - Remove `NumType` and use `float` throughout (author @eseglem, https://github.com/developmentseed/geojson-pydantic/pull/83)

geojson_pydantic/geometries.py

@@ -33,7 +33,8 @@ def __geo_interface__(self):
 class _GeometryBase(BaseModel, GeoInterfaceMixin, abc.ABC):
     """Base class for geometry models"""
 
-    coordinates: Any  # will be constrained in child classes
+    type: str
+    coordinates: Any
 
     @classmethod
     def validate(cls, value):
@@ -61,7 +62,7 @@ def _wkt_inset(self) -> str:
     @property
     def _wkt_type(self) -> str:
         """Return the WKT name of the geometry."""
-        return self.__class__.__name__.upper()
+        return self.type.upper()
 
     @property
     def wkt(self) -> str:
@@ -234,7 +235,7 @@ def __getitem__(self, index):
     @property
     def _wkt_type(self) -> str:
         """Return the WKT name of the geometry."""
-        return self.__class__.__name__.upper()
+        return self.type.upper()
 
     @property
     def _wkt_coordinates(self) -> str:

tests/test_geometries.py

@@ -471,3 +471,14 @@ def test_polygon_from_bounds():
     """Result from `from_bounds` class method should be the same."""
     coordinates = [[(1.0, 2.0), (3.0, 2.0), (3.0, 4.0), (1.0, 4.0), (1.0, 2.0)]]
     assert Polygon(coordinates=coordinates) == Polygon.from_bounds(1.0, 2.0, 3.0, 4.0)
+
+
+def test_wkt_name():
+    """Make sure WKT name is derived from geometry Type."""
+
+    class PointType(Point):
+        ...
+
+    assert (
+        PointType(coordinates=(1.01, 2.01)).wkt == Point(coordinates=(1.01, 2.01)).wkt
+    )