Entire Chesapeake13 test needs to be skipped for missing zipfile_defl…

…ate64
microsoft · Mar 19, 2022 · af38975 · af38975
1 parent a543c24
commit af38975
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/datasets/test_chesapeake.py b/tests/datasets/test_chesapeake.py
@@ -28,9 +28,10 @@ def download_url(url: str, root: str, *args: str, **kwargs: str) -> None:
 
 
 class TestChesapeake13:
+    pytest.importorskip("zipfile_deflate64")
+
     @pytest.fixture
     def dataset(self, monkeypatch: MonkeyPatch, tmp_path: Path) -> Chesapeake13:
-        pytest.importorskip("zipfile_deflate64")
         monkeypatch.setattr(torchgeo.datasets.chesapeake, "download_url", download_url)
         md5 = "fe35a615b8e749b21270472aa98bb42c"
         monkeypatch.setattr(Chesapeake13, "md5", md5)

diff --git a/tests/trainers/test_byol.py b/tests/trainers/test_byol.py
@@ -43,9 +43,6 @@ class TestBYOLTask:
         ],
     )
     def test_trainer(self, name: str, classname: Type[LightningDataModule]) -> None:
-        if name == "chesapeake_cvpr_7":
-            pytest.importorskip("zipfile_deflate64")
-
         conf = OmegaConf.load(os.path.join("conf", "task_defaults", name + ".yaml"))
         conf_dict = OmegaConf.to_object(conf.experiment)
         conf_dict = cast(Dict[Any, Dict[Any, Any]], conf_dict)