diff --git a/tests/test_dataloaders.py b/tests/test_dataloaders.py new file mode 100644 index 0000000..4d23c75 --- /dev/null +++ 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]