Added smoketest for datasets

erikwijmans · Oct 21, 2018 · ca729e7 · ca729e7
1 parent 1b19cbf
commit ca729e7
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_dataloaders.py b/tests/test_dataloaders.py
@@ -0,0 +1,9 @@
+from pointnet2 import data
+
+def test_modelnet40():
+    dset = data.ModelNet40Cls(1024)
+    dset[0]
+
+def test_semseg():
+    dset = data.Indoor3DSemSeg(1024)
+    dset[0]