From 6b61de6847295b7e030c61cab91392723476aa9e Mon Sep 17 00:00:00 2001 From: KevinMusgrave Date: Wed, 11 Dec 2024 18:45:50 +0000 Subject: [PATCH] check if isdir --- tests/datasets/test_cars196.py | 2 +- tests/datasets/test_cub.py | 2 +- tests/datasets/test_inaturalist2018.py | 2 +- tests/datasets/test_sop.py | 3 ++- 4 files changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/datasets/test_cars196.py b/tests/datasets/test_cars196.py index bf1acdca..bd5629f8 100644 --- a/tests/datasets/test_cars196.py +++ b/tests/datasets/test_cars196.py @@ -53,5 +53,5 @@ def test_CARS_196_dataloader(self): @classmethod def tearDownClass(cls): - if not cls.ALREADY_EXISTS: + if not cls.ALREADY_EXISTS and os.path.isdir(cls.CARS_196_ROOT): shutil.rmtree(cls.CARS_196_ROOT) diff --git a/tests/datasets/test_cub.py b/tests/datasets/test_cub.py index 581c15cd..60261965 100644 --- a/tests/datasets/test_cub.py +++ b/tests/datasets/test_cub.py @@ -47,5 +47,5 @@ def test_CUB_dataloader(self): @classmethod def tearDownClass(cls): - if not cls.ALREADY_EXISTS: + if not cls.ALREADY_EXISTS and os.path.isdir(cls.CUB_ROOT): shutil.rmtree(cls.CUB_ROOT) diff --git a/tests/datasets/test_inaturalist2018.py b/tests/datasets/test_inaturalist2018.py index 8f1f6c04..ad26cb7a 100644 --- a/tests/datasets/test_inaturalist2018.py +++ b/tests/datasets/test_inaturalist2018.py @@ -55,5 +55,5 @@ def test_INaturalist2018_dataloader(self): @classmethod def tearDownClass(cls): - if not cls.ALREADY_EXISTS: + if not cls.ALREADY_EXISTS and os.path.isdir(cls.INATURALIST2018_ROOT): shutil.rmtree(cls.INATURALIST2018_ROOT) diff --git a/tests/datasets/test_sop.py b/tests/datasets/test_sop.py index f3788db3..55398716 100644 --- a/tests/datasets/test_sop.py +++ b/tests/datasets/test_sop.py @@ -51,7 +51,8 @@ def test_SOP_dataloader(self): self.assertTupleEqual(tuple(inputs.shape), (8, 3, 224, 224)) self.assertTupleEqual(tuple(labels.shape), (8,)) + @unittest.skipUnless(TEST_DATASETS, "TEST_DATASETS is false") @classmethod def tearDownClass(cls): - if not cls.ALREADY_EXISTS: + if not cls.ALREADY_EXISTS and os.path.isdir(cls.SOP_ROOT): shutil.rmtree(cls.SOP_ROOT)