Cool everything working!

jameschapman19 · Mar 3, 2022 · b847e2e · b847e2e
1 parent 7ab800f
commit b847e2e
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/multiviewdata/test/test_outputs.py b/multiviewdata/test/test_outputs.py
@@ -1,9 +1,26 @@
 import os
 
-from multiviewdata.torchdatasets import NoisyMNISTDataset
-
 
 def test_noisymnist():
+    from multiviewdata.torchdatasets import NoisyMNISTDataset
     a = NoisyMNISTDataset(os.getcwd(), download=True)[0]
     assert "index" in a
     assert "views" in a
+
+def test_tangledmnist():
+    from multiviewdata.torchdatasets import TangledMNISTDataset
+    a = TangledMNISTDataset(os.getcwd(), download=True)[0]
+    assert "index" in a
+    assert "views" in a
+
+def test_splitmnist():
+    from multiviewdata.torchdatasets import SplitMNISTDataset
+    a = SplitMNISTDataset(os.getcwd(), download=True)[0]
+    assert "index" in a
+    assert "views" in a
+
+def test_mfeat():
+    from multiviewdata.torchdatasets import MFeatDataset
+    a = MFeatDataset(os.getcwd(), download=True)[0]
+    assert "index" in a
+    assert "views" in a