Add tests for datasets

torchhd/tests/test_datasets.py

@@ -187,18 +187,16 @@ def test_datasets_dowload(self, teardown):
             if isinstance(cls, type)
         ]
         for dataset_name, dataset_class in dataset_classes:
-            if dataset_name != "UCIHAR":
-                continue
-            dataset = dataset_class("../data", download=True)
-            dataset = dataset_class("../data", download=True)
-
-        # for dataset_name, dataset_class in dataset_classes:
-        #     try:
-        #         if dataset_name == "UCIHAR" or dataset_name == "CyclePowerPlant" or dataset_name == "CollectionDataset":
-        #             continue
-        #         dataset = dataset_class("../data", download=True)
-        #         dataset = dataset_class("../data", download=True)
-        #     except:
-        #         raise RuntimeError(dataset_name)
-        #     assert dataset is not None
-        #     assert len(dataset) > 0
+            try:
+                if (
+                    dataset_name == "UCIHAR"
+                    or dataset_name == "CyclePowerPlant"
+                    or dataset_name == "CollectionDataset"
+                ):
+                    continue
+                dataset = dataset_class("../data", download=True)
+                dataset = dataset_class("../data", download=True)
+            except:
+                raise RuntimeError(dataset_name)
+            assert dataset is not None
+            assert len(dataset) > 0