Fix integration tests on macOS/Windows (#468)

microsoft · Mar 19, 2022 · 26f42b9 · 26f42b9
1 parent a27389f
commit 26f42b9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/samplers/test_batch.py b/tests/samplers/test_batch.py
@@ -51,6 +51,7 @@ def test_len(self, sampler: CustomBatchGeoSampler) -> None:
     @pytest.mark.parametrize("num_workers", [0, 1, 2])
     def test_dataloader(self, sampler: CustomBatchGeoSampler, num_workers: int) -> None:
         ds = CustomGeoDataset()
+        ds.index.insert(0, (0, 100, 200, 300, 400, 500))
         dl = DataLoader(
             ds, batch_sampler=sampler, num_workers=num_workers, collate_fn=stack_samples
         )

diff --git a/tests/samplers/test_single.py b/tests/samplers/test_single.py
@@ -55,6 +55,7 @@ def test_abstract(self) -> None:
     @pytest.mark.parametrize("num_workers", [0, 1, 2])
     def test_dataloader(self, sampler: CustomGeoSampler, num_workers: int) -> None:
         ds = CustomGeoDataset()
+        ds.index.insert(0, (0, 100, 200, 300, 400, 500))
         dl = DataLoader(
             ds, sampler=sampler, num_workers=num_workers, collate_fn=stack_samples
         )