Samplers: fix type hints

Author: adamjstewart

tests/samplers/test_batch.py

@@ -30,7 +30,9 @@ def __iter__(self) -> Iterator[list[BoundingBox]]:
 
 
 class CustomGeoDataset(GeoDataset):
-    def __init__(self, geometry: Sequence[Geometry], res: float = 10) -> None:
+    def __init__(
+        self, geometry: Sequence[Geometry], res: tuple[float, float] = (10, 10)
+    ) -> None:
         intervals = [(MINT, MAXT)] * len(geometry)
         index = pd.IntervalIndex.from_tuples(intervals, closed='both', name='datetime')
         crs = CRS.from_epsg(3005)
@@ -121,7 +123,7 @@ def test_roi(self, dataset: CustomGeoDataset) -> None:
 
     def test_small_area(self) -> None:
         geometry = [shapely.box(0, 0, 10, 10), shapely.box(20, 20, 21, 21)]
-        ds = CustomGeoDataset(geometry, res=1)
+        ds = CustomGeoDataset(geometry, res=(1, 1))
         sampler = RandomBatchGeoSampler(ds, 2, 2, 10)
         for _ in sampler:
             continue

tests/samplers/test_single.py

@@ -37,7 +37,9 @@ def __iter__(self) -> Iterator[BoundingBox]:
 
 
 class CustomGeoDataset(GeoDataset):
-    def __init__(self, geometry: Sequence[Geometry], res: float = 10) -> None:
+    def __init__(
+        self, geometry: Sequence[Geometry], res: tuple[float, float] = (10, 10)
+    ) -> None:
         intervals = [(MINT, MAXT)] * len(geometry)
         index = pd.IntervalIndex.from_tuples(intervals, closed='both', name='datetime')
         crs = CRS.from_epsg(3005)
@@ -114,7 +116,7 @@ def test_roi(self, dataset: CustomGeoDataset) -> None:
 
     def test_small_area(self) -> None:
         geometry = [shapely.box(0, 0, 10, 10), shapely.box(20, 20, 21, 21)]
-        ds = CustomGeoDataset(geometry, res=1)
+        ds = CustomGeoDataset(geometry, res=(1, 1))
         sampler = RandomGeoSampler(ds, 2, 10)
         for _ in sampler:
             continue

tests/samplers/test_utils.py

@@ -2,6 +2,7 @@
 # Licensed under the MIT License.
 
 import math
+from datetime import datetime
 
 import pytest
 
@@ -11,6 +12,9 @@
 
 MAYBE_TUPLE = float | tuple[float, float]
 
+MINT = datetime(2025, 4, 24)
+MAXT = datetime(2025, 4, 25)
+
 
 @pytest.mark.parametrize(
     'size,stride,expected',
@@ -34,7 +38,7 @@
 def test_tile_to_chips(
     size: MAYBE_TUPLE, stride: MAYBE_TUPLE | None, expected: MAYBE_TUPLE
 ) -> None:
-    bounds = BoundingBox(0, 10, 20, 30, 40, 50)
+    bounds = BoundingBox(0, 10, 20, 30, MINT, MAXT)
     size = _to_tuple(size)
     if stride is not None:
         stride = _to_tuple(stride)