From ca729e7f3f4e3b22ea0f8206765f3a20e73c522c Mon Sep 17 00:00:00 2001 From: erikwijmans Date: Sun, 21 Oct 2018 13:42:00 -0400 Subject: [PATCH] Added smoketest for datasets --- tests/test_dataloaders.py | 9 +++++++++ 1 file changed, 9 insertions(+) create mode 100644 tests/test_dataloaders.py 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]