Unverified Commit 1826e92d authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

add tests for Kinetics400 (#3457)



* add tests for Kinetics400

* use create_random_string()
Co-authored-by: default avatarPhilip Meier <pmeier.github@posteo.de>
Co-authored-by: default avatarFrancisco Massa <fvsmassa@gmail.com>
parent a7a2ee7c
......@@ -1003,6 +1003,34 @@ class LSUNTestCase(datasets_utils.ImageDatasetTestCase):
return num_images
def test_not_found_or_corrupted(self):
# LSUN does not raise built-in exception, but a custom one. It is expressive enough to not 'cast' it to
# RuntimeError or FileNotFoundError that are normally checked by this test.
with self.assertRaises(datasets_utils.lazy_importer.lmdb.Error):
super().test_not_found_or_corrupted()
class Kinetics400TestCase(datasets_utils.VideoDatasetTestCase):
DATASET_CLASS = datasets.Kinetics400
def inject_fake_data(self, tmpdir, config):
classes = ("Abseiling", "Zumba")
num_videos_per_class = 2
digits = string.ascii_letters + string.digits + "-_"
for cls in classes:
datasets_utils.create_video_folder(
tmpdir,
cls,
lambda _: f"{datasets_utils.create_random_string(11, digits)}.avi",
num_videos_per_class,
)
return num_videos_per_class * len(classes)
def test_not_found_or_corrupted(self):
self.skipTest("Dataset currently does not handle the case of no found videos.")
if __name__ == "__main__":
unittest.main()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment