Skip to content

Commit 9846569

Browse files
pmeierdatumbox
andauthored
add tests for FakeData dataset (#3468)
* add tests for FakeData dataset * improve skip reason Co-authored-by: Vasilis Vryniotis <[email protected]>
1 parent 4fcaee0 commit 9846569

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

test/test_datasets.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1261,5 +1261,19 @@ def _file_stem(self, idx):
12611261
return f"2008_{idx:06d}"
12621262

12631263

1264+
class FakeDataTestCase(datasets_utils.ImageDatasetTestCase):
1265+
DATASET_CLASS = datasets.FakeData
1266+
FEATURE_TYPES = (PIL.Image.Image, torch.Tensor)
1267+
1268+
def dataset_args(self, tmpdir, config):
1269+
return ()
1270+
1271+
def inject_fake_data(self, tmpdir, config):
1272+
return config["size"]
1273+
1274+
def test_not_found_or_corrupted(self):
1275+
self.skipTest("The data is generated at creation and thus cannot be non-existent or corrupted.")
1276+
1277+
12641278
if __name__ == "__main__":
12651279
unittest.main()

0 commit comments

Comments
 (0)