Fix integration tests on macOS/Windows (microsoft#468)

yichiac · Mar 19, 2022 · 457fc81 · 457fc81
1 parent 53cfecc
commit 457fc81
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/tests/samplers/test_batch.py b/tests/samplers/test_batch.py
@@ -39,7 +39,9 @@ def __getitem__(self, query: BoundingBox) -> Dict[str, BoundingBox]:
 class TestBatchGeoSampler:
     @pytest.fixture(scope="class")
     def dataset(self) -> CustomGeoDataset:
-        return CustomGeoDataset()
+        ds = CustomGeoDataset()
+        ds.index.insert(0, (0, 100, 200, 300, 400, 500))
+        return ds
 
     @pytest.fixture(scope="function")
     def sampler(self) -> CustomBatchGeoSampler:

diff --git a/tests/samplers/test_single.py b/tests/samplers/test_single.py
@@ -39,7 +39,9 @@ def __getitem__(self, query: BoundingBox) -> Dict[str, BoundingBox]:
 class TestGeoSampler:
     @pytest.fixture(scope="class")
     def dataset(self) -> CustomGeoDataset:
-        return CustomGeoDataset()
+        ds = CustomGeoDataset()
+        ds.index.insert(0, (0, 100, 200, 300, 400, 500))
+        return ds
 
     @pytest.fixture(scope="function")
     def sampler(self) -> CustomGeoSampler: