Skip to content

Commit

Permalink
refactor: rename reset_extents to reset_extent for consistency
Browse files Browse the repository at this point in the history
  • Loading branch information
paulfouquet committed Feb 25, 2025
1 parent d17e78b commit 8d8d481
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions scripts/stac/imagery/collection.py
Original file line number Diff line number Diff line change
Expand Up @@ -247,13 +247,13 @@ def update_extent(self, bbox: list[float] | None = None, interval: list[str] | N
interval: datetime interval. Defaults to None.
"""
if "extent" not in self.stac:
self.reset_extents()
self.reset_extent()
if bbox:
self.stac["extent"]["spatial"]["bbox"] = [bbox]
if interval:
self.stac["extent"]["temporal"]["interval"] = [interval]

def reset_extents(self) -> None:
def reset_extent(self) -> None:
"""Reset the spatial and temporal extents of the Collection."""
self.stac.setdefault("extent", {}).setdefault("spatial", {})["bbox"] = None
self.stac.setdefault("extent", {}).setdefault("temporal", {})["interval"] = None
Expand Down
4 changes: 2 additions & 2 deletions scripts/stac/imagery/tests/collection_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -508,12 +508,12 @@ def test_capture_dates_added(fake_collection_metadata: CollectionMetadata, fake_
}


def test_reset_extents(fake_collection_metadata: CollectionMetadata, fake_linz_slug: str) -> None:
def test_reset_extent(fake_collection_metadata: CollectionMetadata, fake_linz_slug: str) -> None:
collection = ImageryCollection(
fake_collection_metadata, any_epoch_datetime_string(), any_epoch_datetime_string(), fake_linz_slug
)
collection.update_temporal_extent("2021-01-27T00:00:00Z", "2021-01-27T00:00:00Z")
collection.update_spatial_extent([1799667.5, 5815977.0, 1800422.5, 5814986.0])
collection.reset_extents()
collection.reset_extent()
assert collection.stac["extent"]["spatial"]["bbox"] is None
assert collection.stac["extent"]["temporal"]["interval"] is None

0 comments on commit 8d8d481

Please sign in to comment.