Skip tests that require optional deps

microsoft · Sep 30, 2023 · da9783f · da9783f
1 parent e3b3544
commit da9783f
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 0 deletions.
diff --git a/tests/datasets/test_advance.py b/tests/datasets/test_advance.py
@@ -81,6 +81,7 @@ def test_mock_missing_module(
             dataset[0]
 
     def test_plot(self, dataset: ADVANCE) -> None:
+        pytest.importorskip("scipy", minversion="1.6.2")
         x = dataset[0].copy()
         dataset.plot(x, suptitle="Test")
         plt.close()

diff --git a/tests/datasets/test_skippd.py b/tests/datasets/test_skippd.py
@@ -19,6 +19,9 @@
 from torchgeo.datasets import SKIPPD
 
 
+pytest.importorskip("h5py", minversion="3")
+
+
 def download_url(url: str, root: str, *args: str, **kwargs: str) -> None:
     shutil.copy(url, root)
 

diff --git a/tests/trainers/test_detection.py b/tests/trainers/test_detection.py
@@ -18,6 +18,10 @@
 from torchgeo.trainers import ObjectDetectionTask
 
 
+# MAP metric requires pycocotools to be installed
+pytest.importorskip("pycocotools")
+
+
 class PredictObjectDetectionDataModule(NASAMarineDebrisDataModule):
     def setup(self, stage: str) -> None:
         self.predict_dataset = NASAMarineDebris(**self.kwargs)

diff --git a/tests/trainers/test_regression.py b/tests/trainers/test_regression.py
@@ -66,6 +66,9 @@ def create_model(*args: Any, **kwargs: Any) -> Module:
     def test_trainer(
         self, monkeypatch: MonkeyPatch, name: str, fast_dev_run: bool
     ) -> None:
+        if name == "skippd":
+            pytest.importorskip("h5py", minversion="3")
+
         config = os.path.join("tests", "conf", name + ".yaml")
 
         monkeypatch.setattr(timm, "create_model", self.create_model)