fix datamodules failing test, better test for resampling

sede-open · Sep 18, 2024 · 6fba6cc · 6fba6cc
1 parent 83411f4
commit 6fba6cc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tests/datamodules/test_geo.py b/tests/datamodules/test_geo.py
@@ -11,6 +11,7 @@
 from matplotlib.figure import Figure
 from rasterio.crs import CRS
 from torch import Tensor
+from geopandas import GeoDataFrame
 
 from torchgeo.datamodules import (
     GeoDataModule,
@@ -182,7 +183,7 @@ def test_zero_length_sampler(self) -> None:
         dm = CustomGeoDataModule()
         dm.dataset = CustomGeoDataset()
         dm.sampler = RandomGeoSampler(dm.dataset, 1, 1)
-        dm.sampler.length = 0
+        dm.sampler.chips = GeoDataFrame()
         msg = r'CustomGeoDataModule\.sampler has length 0.'
         with pytest.raises(MisconfigurationException, match=msg):
             dm.train_dataloader()

diff --git a/tests/samplers/test_single.py b/tests/samplers/test_single.py
@@ -190,6 +190,7 @@ def test_empty(self, dataset: CustomGeoDataset) -> None:
         assert len(sampler) == 0
 
     def test_refresh_samples(self, dataset: CustomGeoDataset) -> None:
+        dataset.index.insert(0, (0, 100, 200, 300, 400, 500))
         sampler = RandomGeoSampler(dataset, 5, length=1)
         samples = list(sampler)
         assert len(sampler) == 1