Added download test for KITTI dataset (#3736)

7bc5bd85 · Prabhat Roy · GitHub · 5bc68cc9 · 7bc5bd85
Unverified Commit 7bc5bd85 authored Apr 26, 2021 by Prabhat Roy Committed by GitHub Apr 26, 2021
Show whitespace changes
Inline Side-by-side

Showing with 14 additions and 0 deletions

test/test_datasets_download.py test/test_datasets_download.py +14 -0

No files found.
--- a/test/test_datasets_download.py
+++ b/test/test_datasets_download.py
@@ -392,6 +392,19 @@ def widerface():
    )


+def kitti():
+    return itertools.chain(
+        *[
+            collect_download_configs(
+                lambda train=train: datasets.Kitti(ROOT, train=train, download=True),
+                name=f"Kitti, {'train' if train else 'test'}",
+                file="kitti",
+            )
+            for train in (True, False)
+        ]
+    )
+
+
 def make_parametrize_kwargs(download_configs):
    argvalues = []
    ids = []
@@ -427,6 +440,7 @@ def make_parametrize_kwargs(download_configs):
            usps(),
            celeba(),
            widerface(),
+            kitti(),
        )
    )
 )