361 test geometries with hypothesis

.github/workflows/run-all-tests.yml

@@ -2,7 +2,7 @@
 # run tests and lint with a variety of Python versions
 ---
 name: Tests
 on: [push, pull_request]
 
 jobs:
   cpython:
@@ -43,7 +43,7 @@
           pip install -e ".[tests, lxml]"
       - name: Test with pytest
         run: |
-          pytest tests --cov=fastkml --cov=tests --cov-fail-under=88 --cov-report=xml
+          pytest tests --cov=fastkml --cov=tests --cov-fail-under=95 --cov-report=xml
       - name: "Upload coverage to Codecov"
         if: ${{ matrix.python-version==3.11 }}
         uses: codecov/codecov-action@v4
@@ -127,6 +127,9 @@
     needs: [cpython, static-tests, pypy, cpython-lxml, doctest-lxml]
     name: Build and publish to PyPI and TestPyPI
     runs-on: ubuntu-latest
+    environment: release
+    permissions:
+      id-token: write
     steps:
       - uses: actions/checkout@v4
       - name: Set up Python
@@ -150,11 +153,8 @@
         if: startsWith(github.ref, 'refs/tags')
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
-          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
           repository_url: https://test.pypi.org/legacy/
       - name: Publish distribution 📦 to PyPI for push to main
         if: github.event_name == 'push' && github.ref == 'refs/heads/main'
         uses: pypa/gh-action-pypi-publish@release/v1
-        with:
-          password: ${{ secrets.PYPI_API_TOKEN }}
 ...

.pre-commit-config.yaml

@@ -40,7 +40,7 @@ repos:
     rev: v3.17.0
     hooks:
       - id: pyupgrade
-        args: ["--py3-plus", "--py37-plus"]
+        args: ["--py3-plus", "--py38-plus"]
   - repo: https://github.com/psf/black
     rev: 24.8.0
     hooks:
@@ -77,7 +77,7 @@ repos:
     rev: v1.11.2
     hooks:
       - id: mypy
-        additional_dependencies: [pygeoif>=1.4, arrow, pytest]
+        additional_dependencies: [pygeoif>=1.4, arrow, pytest, hypothesis]
   - repo: https://github.com/adamchainz/blacken-docs
     rev: "1.18.0"
     hooks:

fastkml/features.py

@@ -185,6 +185,7 @@ def __bool__(self) -> bool:
         classes=(int,),
         get_kwarg=attribute_int_kwarg,
         set_element=int_attribute,
+        default=2,
     ),
 )
 

fastkml/geometry.py

@@ -93,6 +93,8 @@
 
 MsgMutualExclusive: Final = "Geometry and kml coordinates are mutually exclusive"
 
+xml_attrs = {"ns", "name_spaces", "id", "target_id"}
+
 
 def handle_invalid_geometry_error(
     *,
@@ -159,13 +161,12 @@
 
     """
     if getattr(obj, attr_name, None):
-        p = precision if precision is not None else 6
         coords = getattr(obj, attr_name)
-        if len(coords[0]) == 2:  # noqa: PLR2004
-            tuples = (f"{c[0]:.{p}f},{c[1]:.{p}f}" for c in coords)
-        elif len(coords[0]) == 3:  # noqa: PLR2004
-            tuples = (f"{c[0]:.{p}f},{c[1]:.{p}f},{c[2]:.{p}f}" for c in coords)
+        if precision is None:
+            tuples = (",".join(str(c) for c in coord) for coord in coords)
         else:
+            tuples = (",".join(f"{c:.{precision}f}" for c in coord) for coord in coords)
+        if len(coords[0]) not in (2, 3):
             msg = f"Invalid dimensions in coordinates '{coords}'"
             raise KMLWriteError(msg)
         element.text = " ".join(tuples)
@@ -488,6 +489,21 @@
         """
         return bool(self.geometry)
 
+    def __eq__(self, other: object) -> bool:
+        """Check if the Point objects are equal."""
+        if isinstance(other, Point):
+            return all(
+                getattr(self, attr) == getattr(other, attr)
+                for attr in (
+                    "extrude",
+                    "altitude_mode",
+                    "geometry",
+                    *xml_attrs,
+                    *self._get_splat(),
+                )
+            )
+        return super().__eq__(other)
+
     @property
     def geometry(self) -> Optional[geo.Point]:
         """
@@ -628,6 +644,21 @@
         """
         return bool(self.geometry)
 
+    def __eq__(self, other: object) -> bool:
+        """Check if the LineString objects is equal."""
+        if isinstance(other, LineString):
+            return all(
+                getattr(self, attr) == getattr(other, attr)
+                for attr in (
+                    "extrude",
+                    "tessellate",
+                    "geometry",
+                    *xml_attrs,
+                    *self._get_splat(),
+                )
+            )
+        return super().__eq__(other)
+
     @property
     def geometry(self) -> Optional[geo.LineString]:
         """
@@ -754,22 +785,6 @@
             **kwargs,
         )
 
-    def __repr__(self) -> str:
-        """Create a string (c)representation for LinearRing."""
-        return (
-            f"{self.__class__.__module__}.{self.__class__.__name__}("
-            f"ns={self.ns!r}, "
-            f"name_spaces={self.name_spaces!r}, "
-            f"id={self.id!r}, "
-            f"target_id={self.target_id!r}, "
-            f"extrude={self.extrude!r}, "
-            f"tessellate={self.tessellate!r}, "
-            f"altitude_mode={self.altitude_mode}, "
-            f"geometry={self.geometry!r}, "
-            f"**{self._get_splat()!r},"
-            ")"
-        )
-
     @property
     def geometry(self) -> Optional[geo.LinearRing]:
         """
@@ -1182,12 +1197,26 @@
             f"extrude={self.extrude!r}, "
             f"tessellate={self.tessellate!r}, "
             f"altitude_mode={self.altitude_mode}, "
-            f"outer_boundary={self.outer_boundary!r}, "
-            f"inner_boundaries={self.inner_boundaries!r}, "
+            f"geometry={self.geometry!r}, "
             f"**{self._get_splat()!r},"
             ")"
         )
 
+    def __eq__(self, other: object) -> bool:
+        """Check if the Polygon objects are equal."""
+        if isinstance(other, Polygon):
+            return all(
+                getattr(self, attr) == getattr(other, attr)
+                for attr in (
+                    "extrude",
+                    "tessellate",
+                    "geometry",
+                    *xml_attrs,
+                    *self._get_splat(),
+                )
+            )
+        return super().__eq__(other)
+
 
 registry.register(
     Polygon,

pyproject.toml

@@ -77,6 +77,7 @@ lxml = [
     "lxml",
 ]
 tests = [
+    "hypothesis",
     "pytest",
     "pytest-cov",
 ]
@@ -117,6 +118,7 @@ source = [
 [tool.coverage.report]
 exclude_also = [
     "^\\s*\\.\\.\\.$",
+    "class \\w+\\(Protocol\\)\\:",
     "except AssertionError:",
     "except ImportError:",
     "if TYPE_CHECKING:",

tests/conftest.py

@@ -0,0 +1,21 @@
+"""
+Configure the tests.
+
+Register the hypothesis 'exhaustive' profile to run 10 thousand examples.
+Run this profile with ``pytest --hypothesis-profile=exhaustive``
+"""
+
+from hypothesis import HealthCheck
+from hypothesis import settings
+
+settings.register_profile(
+    "exhaustive",
+    max_examples=10_000,
+    suppress_health_check=[HealthCheck.too_slow],
+)
+settings.register_profile(
+    "coverage",
+    max_examples=10,
+    suppress_health_check=[HealthCheck.too_slow],
+)
+settings.register_profile("ci", suppress_health_check=[HealthCheck.too_slow])

tests/geometries/boundaries_test.py

@@ -35,7 +35,7 @@ def test_outer_boundary(self) -> None:
         )
 
         assert outer_boundary.geometry == geo.LinearRing(coords)
-        assert outer_boundary.to_string(prettyprint=False).strip() == (
+        assert outer_boundary.to_string(prettyprint=False, precision=6).strip() == (
             '<kml:outerBoundaryIs xmlns:kml="http://www.opengis.net/kml/2.2">'
             "<kml:LinearRing><kml:coordinates>"
             "1.000000,2.000000 2.000000,0.000000 0.000000,0.000000 1.000000,2.000000"
@@ -66,7 +66,7 @@ def test_inner_boundary(self) -> None:
 
         assert inner_boundary.geometry == geo.LinearRing(coords)
         assert bool(inner_boundary)
-        assert inner_boundary.to_string(prettyprint=False).strip() == (
+        assert inner_boundary.to_string(prettyprint=False, precision=6).strip() == (
             '<kml:innerBoundaryIs xmlns:kml="http://www.opengis.net/kml/2.2">'
             "<kml:LinearRing><kml:coordinates>"
             "1.000000,2.000000 2.000000,0.000000 0.000000,0.000000 1.000000,2.000000"

tests/geometries/coordinates_test.py

@@ -29,7 +29,7 @@ def test_coordinates(self) -> None:
 
         coordinates = Coordinates(coords=coords)
 
-        assert coordinates.to_string().strip() == (
+        assert coordinates.to_string(precision=6).strip() == (
             '<kml:coordinates xmlns:kml="http://www.opengis.net/kml/2.2">'
             "0.000000,0.000000 0.000000,1.000000 1.000000,1.000000 "
             "1.000000,0.000000 0.000000,0.000000"

tests/geometries/geometry_test.py

@@ -495,7 +495,7 @@ def test_create_kml_geometry_point(self) -> None:
             "coordinates": (0.0, 1.0),
         }
         assert "Point>" in g.to_string()
-        assert "coordinates>0.000000,1.000000</" in g.to_string()
+        assert "coordinates>0.000000,1.000000</" in g.to_string(precision=6)
 
     def test_create_kml_geometry_linestring(self) -> None:
         """Test the create_kml_geometry function."""
@@ -509,7 +509,9 @@ def test_create_kml_geometry_linestring(self) -> None:
             "coordinates": ((0.0, 0.0), (1.0, 1.0)),
         }
         assert "LineString>" in g.to_string()
-        assert "coordinates>0.000000,0.000000 1.000000,1.000000</" in g.to_string()
+        assert "coordinates>0.000000,0.000000 1.000000,1.000000</" in g.to_string(
+            precision=6,
+        )
 
     def test_create_kml_geometry_linearring(self) -> None:
         """Test the create_kml_geometry function."""
@@ -526,7 +528,7 @@ def test_create_kml_geometry_linearring(self) -> None:
         assert (
             "coordinates>0.000000,0.000000 1.000000,1.000000 1.000000,0.000000 "
             "0.000000,0.000000</"
-        ) in g.to_string()
+        ) in g.to_string(precision=6)
 
     def test_create_kml_geometry_polygon(self) -> None:
         """Test the create_kml_geometry function."""
@@ -543,7 +545,7 @@ def test_create_kml_geometry_polygon(self) -> None:
         assert (
             "coordinates>0.000000,0.000000 1.000000,1.000000 1.000000,0.000000 "
             "0.000000,0.000000</"
-        ) in g.to_string()
+        ) in g.to_string(precision=6)
 
     def test_create_kml_geometry_multipoint(self) -> None:
         """Test the create_kml_geometry function."""
@@ -552,12 +554,13 @@ def test_create_kml_geometry_multipoint(self) -> None:
         assert isinstance(g, MultiGeometry)
         assert g.geometry
         assert len(g.geometry) == 4
-        assert "MultiGeometry>" in g.to_string()
-        assert "Point>" in g.to_string()
-        assert "coordinates>0.000000,0.000000</" in g.to_string()
-        assert "coordinates>1.000000,1.000000</" in g.to_string()
-        assert "coordinates>1.000000,0.000000</" in g.to_string()
-        assert "coordinates>2.000000,2.000000</" in g.to_string()
+        xml = g.to_string(precision=6)
+        assert "MultiGeometry>" in xml
+        assert "Point>" in xml
+        assert "coordinates>0.000000,0.000000</" in xml
+        assert "coordinates>1.000000,1.000000</" in xml
+        assert "coordinates>1.000000,0.000000</" in xml
+        assert "coordinates>2.000000,2.000000</" in xml
 
     def test_create_kml_geometry_multilinestring(self) -> None:
         """Test the create_kml_geometry function."""
@@ -568,10 +571,11 @@ def test_create_kml_geometry_multilinestring(self) -> None:
         assert isinstance(g, MultiGeometry)
         assert g.geometry
         assert len(g.geometry) == 2
-        assert "MultiGeometry>" in g.to_string()
-        assert "LineString>" in g.to_string()
-        assert "coordinates>0.000000,0.000000 1.000000,1.000000</" in g.to_string()
-        assert "coordinates>0.000000,0.000000 1.000000,1.000000</" in g.to_string()
+        xml = g.to_string(precision=6)
+        assert "MultiGeometry>" in xml
+        assert "LineString>" in xml
+        assert "coordinates>0.000000,0.000000 1.000000,1.000000</" in xml
+        assert "coordinates>0.000000,0.000000 1.000000,1.000000</" in xml
 
     def test_create_kml_geometry_multipolygon(self) -> None:
         """Test the create_kml_geometry function."""
@@ -590,20 +594,21 @@ def test_create_kml_geometry_multipolygon(self) -> None:
         assert isinstance(g, MultiGeometry)
         assert g.geometry
         assert len(g.geometry) == 2
-        assert "MultiGeometry>" in g.to_string()
-        assert "Polygon>" in g.to_string()
+        xml = g.to_string(precision=6)
+        assert "MultiGeometry>" in xml
+        assert "Polygon>" in xml
         assert (
             "coordinates>0.000000,0.000000 0.000000,1.000000 1.000000,1.000000 "
             "1.000000,0.000000 0.000000,0.000000</"
-        ) in g.to_string()
+        ) in xml
         assert (
             "coordinates>0.100000,0.100000 0.100000,0.200000 0.200000,0.200000 "
             "0.200000,0.100000 0.100000,0.100000</"
-        ) in g.to_string()
+        ) in xml
         assert (
             "coordinates>0.000000,0.000000 0.000000,1.000000 1.000000,1.000000 "
             "1.000000,0.000000 0.000000,0.000000</"
-        ) in g.to_string()
+        ) in xml
 
     def test_create_kml_geometry_geometrycollection(self) -> None:
         multipoint = geo.MultiPoint([(0, 0), (1, 1), (1, 2), (2, 2)])
@@ -625,14 +630,15 @@ def test_create_kml_geometry_geometrycollection(self) -> None:
         assert isinstance(g, MultiGeometry)
         assert g.geometry
         assert len(g.geometry) == 7
-        assert "MultiGeometry>" in g.to_string()
-        assert "LineString>" in g.to_string()
-        assert "LinearRing>" in g.to_string()
-        assert "Polygon>" in g.to_string()
-        assert "outerBoundaryIs>" in g.to_string()
-        assert "innerBoundaryIs>" in g.to_string()
-        assert "Point>" in g.to_string()
-        assert "coordinates>0.000000,0.000000</" in g.to_string()
+        xml = g.to_string(precision=6)
+        assert "MultiGeometry>" in xml
+        assert "LineString>" in xml
+        assert "LinearRing>" in xml
+        assert "Polygon>" in xml
+        assert "outerBoundaryIs>" in xml
+        assert "innerBoundaryIs>" in xml
+        assert "Point>" in xml
+        assert "coordinates>0.000000,0.000000</" in xml
         assert g.geometry == gc
 
     def test_create_kml_geometry_geometrycollection_roundtrip(self) -> None:

tests/geometries/linearring_test.py

@@ -45,7 +45,8 @@ def test_to_string(self) -> None:
         assert "LinearRing" in linear_ring.to_string()
         assert (
             "coordinates>0.000000,0.000000 0.000000,1.000000 1.000000,1.000000 "
-            "1.000000,0.000000 0.000000,0.000000</" in linear_ring.to_string()
+            "1.000000,0.000000 0.000000,0.000000</"
+            in linear_ring.to_string(precision=6)
         )
 
     def test_from_string(self) -> None:

tests/geometries/linestring_test.py

@@ -47,7 +47,7 @@ def test_to_string(self) -> None:
         assert "LineString" in line_string.to_string()
         assert (
             "coordinates>1.000000,2.000000 2.000000,0.000000</"
-            in line_string.to_string()
+            in line_string.to_string(precision=6)
         )
 
     def test_from_string(self) -> None: