remove conflicting iter len getitem methods

Author: vincentsarago

CHANGELOG.md

@@ -8,6 +8,71 @@ Note: Minor version `0.X.0` update might break the API, It's recommended to pin
 
 ## [unreleased]
 
+## [2.0.0] - TBD
+
+* remove custom `__iter__`, `__getitem__` and `__len__` methods from `GeometryCollection` class  **breaking change**
+
+    ```python
+    from geojson_pydantic.geometries import GeometryCollection, Point, MultiPoint
+
+    geoms = GeometryCollection(
+        type="GeometryCollection",
+        geometries=[
+            Point(type="Point", coordinates=(102.0, 0.5)),
+            MultiPoint(type="MultiPoint", coordinates=[(100.0, 0.0), (101.0, 1.0)]),
+        ],
+    )
+
+    ########
+    # Before
+    for geom in geom:       # __iter__
+        pass
+
+    assert len(geoms) == 2  # __len__
+
+    _ = geoms[0]            # __getitem__
+
+    #####
+    # Now
+    for geom in geom.iter():   # __iter__
+        pass
+
+    assert geoms.length == 2  # __len__
+
+    _ = geoms.geometries[0]   # __getitem__
+    ```
+
+* remove custom `__iter__`, `__getitem__` and `__len__` methods from `FeatureCollection` class  **breaking change**
+
+    ```python
+    from geojson_pydantic import FeatureCollection, Feature, Point
+
+    fc = FeatureCollection(
+        type="FeatureCollection", features=[
+            Feature(type="Feature", geometry=Point(type="Point", coordinates=(102.0, 0.5)), properties={"name": "point1"}),
+            Feature(type="Feature", geometry=Point(type="Point", coordinates=(102.0, 1.5)), properties={"name": "point2"}),
+        ]
+    )
+
+    ########
+    # Before
+    for feat in fc:      # __iter__
+        pass
+
+    assert len(fc) == 2  # __len__
+
+    _ = fc[0]            # __getitem__
+
+    #####
+    # Now
+    for feat in fc.iter(): # __iter__
+        pass
+
+    assert fc.length == 2  # __len__
+
+    _ = fe.features[0]     # __getitem__
+    ```
+
 ## [1.2.0] - 2024-12-19
 
 * drop python 3.8 support
@@ -376,7 +441,8 @@ Although the type file was added in `0.2.0` it wasn't included in the distribute
 ### Added
 - Initial Release
 
-[unreleased]: https://github.com/developmentseed/geojson-pydantic/compare/1.2.0...HEAD
+[unreleased]: https://github.com/developmentseed/geojson-pydantic/compare/2.0.0...HEAD
+[2.0.0]: https://github.com/developmentseed/geojson-pydantic/compare/1.2.0...2.0.0
 [1.2.0]: https://github.com/developmentseed/geojson-pydantic/compare/1.1.2...1.2.0
 [1.1.2]: https://github.com/developmentseed/geojson-pydantic/compare/1.1.1...1.1.2
 [1.1.1]: https://github.com/developmentseed/geojson-pydantic/compare/1.1.0...1.1.1

geojson_pydantic/features.py

@@ -39,14 +39,11 @@ class FeatureCollection(_GeoJsonBase, Generic[Feat]):
     type: Literal["FeatureCollection"]
     features: List[Feat]
 
-    def __iter__(self) -> Iterator[Feat]:  # type: ignore [override]
+    def iter(self) -> Iterator[Feat]:
         """iterate over features"""
         return iter(self.features)
 
-    def __len__(self) -> int:
+    @property
+    def length(self) -> int:
         """return features length"""
         return len(self.features)
-
-    def __getitem__(self, index: int) -> Feat:
-        """get feature at a given index"""
-        return self.features[index]

geojson_pydantic/geometries.py

@@ -251,18 +251,15 @@ class GeometryCollection(_GeoJsonBase):
     type: Literal["GeometryCollection"]
     geometries: List[Geometry]
 
-    def __iter__(self) -> Iterator[Geometry]:  # type: ignore [override]
+    def iter(self) -> Iterator[Geometry]:
         """iterate over geometries"""
         return iter(self.geometries)
 
-    def __len__(self) -> int:
+    @property
+    def length(self) -> int:
         """return geometries length"""
         return len(self.geometries)
 
-    def __getitem__(self, index: int) -> Geometry:
-        """get geometry at a given index"""
-        return self.geometries[index]
-
     @property
     def wkt(self) -> str:
         """Return the Well Known Text representation."""

tests/test_features.py

@@ -90,7 +90,9 @@ def test_feature_collection_iteration():
         type="FeatureCollection", features=[test_feature, test_feature]
     )
     assert hasattr(gc, "__geo_interface__")
-    iter(gc)
+    assert list(iter(gc))
+    assert len(list(gc.iter())) == 2
+    assert dict(gc)
 
 
 def test_geometry_collection_iteration():
@@ -99,7 +101,9 @@ def test_geometry_collection_iteration():
         type="FeatureCollection", features=[test_feature_geometry_collection]
     )
     assert hasattr(gc, "__geo_interface__")
-    iter(gc)
+    assert list(iter(gc))
+    assert len(list(gc.iter())) == 1
+    assert dict(gc)
 
 
 def test_generic_properties_is_dict():
@@ -177,9 +181,11 @@ def test_feature_collection_generic():
     fc = FeatureCollection[Feature[Polygon, GenericProperties]](
         type="FeatureCollection", features=[test_feature, test_feature]
     )
-    assert len(fc) == 2
-    assert type(fc[0].properties) == GenericProperties
-    assert type(fc[0].geometry) == Polygon
+    assert fc.length == 2
+    assert len(list(fc.iter())) == 2
+    assert type(fc.features[0].properties) == GenericProperties
+    assert type(fc.features[0].geometry) == Polygon
+    assert dict(fc)
 
 
 def test_geo_interface_protocol():

tests/test_geometries.py

@@ -498,7 +498,8 @@ def test_geometry_collection_iteration(coordinates):
         type="GeometryCollection", geometries=[polygon, multipolygon]
     )
     assert hasattr(gc, "__geo_interface__")
-    iter(gc)
+    assert len(list(gc.iter())) == 2
+    assert list(iter(gc))
 
 
 @pytest.mark.parametrize(
@@ -511,7 +512,10 @@ def test_len_geometry_collection(coordinates):
     gc = GeometryCollection(
         type="GeometryCollection", geometries=[polygon, multipolygon]
     )
-    assert len(gc) == 2
+    assert gc.length == 2
+    assert len(list(gc.iter())) == 2
+    assert dict(gc)
+    assert list(iter(gc))
 
 
 @pytest.mark.parametrize(
@@ -524,8 +528,8 @@ def test_getitem_geometry_collection(coordinates):
     gc = GeometryCollection(
         type="GeometryCollection", geometries=[polygon, multipolygon]
     )
-    assert polygon == gc[0]
-    assert multipolygon == gc[1]
+    assert polygon == gc.geometries[0]
+    assert multipolygon == gc.geometries[1]
 
 
 def test_wkt_mixed_geometry_collection():