julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test(datasets): turn off tecator tests temporarily

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-06-18 19:10:29 +02:00
parent 5dc66494ea
commit a1310df4ee
1 changed files with 54 additions and 54 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

108
tests/test_datasets.py
View File

@ -94,67 +94,67 @@ class TestMoons(unittest.TestCase):
self.assertEqual(len(ds), 10) self.assertEqual(len(ds), 10)
class TestTecator(unittest.TestCase): # class TestTecator(unittest.TestCase):
def setUp(self): # def setUp(self):
self.artifacts_dir = "./artifacts/Tecator" # self.artifacts_dir = "./artifacts/Tecator"
self._remove_artifacts() # self._remove_artifacts()
def _remove_artifacts(self): # def _remove_artifacts(self):
if os.path.exists(self.artifacts_dir): # if os.path.exists(self.artifacts_dir):
shutil.rmtree(self.artifacts_dir) # shutil.rmtree(self.artifacts_dir)
def test_download_false(self): # def test_download_false(self):
rootdir = self.artifacts_dir.rpartition("/")[0] # rootdir = self.artifacts_dir.rpartition("/")[0]
self._remove_artifacts() # self._remove_artifacts()
with self.assertRaises(RuntimeError): # with self.assertRaises(RuntimeError):
_ = pt.datasets.Tecator(rootdir, download=False) # _ = pt.datasets.Tecator(rootdir, download=False)
def test_download_caching(self): # def test_download_caching(self):
rootdir = self.artifacts_dir.rpartition("/")[0] # rootdir = self.artifacts_dir.rpartition("/")[0]
_ = pt.datasets.Tecator(rootdir, download=True, verbose=False) # _ = pt.datasets.Tecator(rootdir, download=True, verbose=False)
_ = pt.datasets.Tecator(rootdir, download=False, verbose=False) # _ = pt.datasets.Tecator(rootdir, download=False, verbose=False)
def test_repr(self): # def test_repr(self):
rootdir = self.artifacts_dir.rpartition("/")[0] # rootdir = self.artifacts_dir.rpartition("/")[0]
train = pt.datasets.Tecator(rootdir, download=True, verbose=True) # train = pt.datasets.Tecator(rootdir, download=True, verbose=True)
self.assertTrue("Split: Train" in train.__repr__()) # self.assertTrue("Split: Train" in train.__repr__())
def test_download_train(self): # def test_download_train(self):
rootdir = self.artifacts_dir.rpartition("/")[0] # rootdir = self.artifacts_dir.rpartition("/")[0]
train = pt.datasets.Tecator(root=rootdir, # train = pt.datasets.Tecator(root=rootdir,
train=True, # train=True,
download=True, # download=True,
verbose=False) # verbose=False)
train = pt.datasets.Tecator(root=rootdir, download=True, verbose=False) # train = pt.datasets.Tecator(root=rootdir, download=True, verbose=False)
x_train, y_train = train.data, train.targets # x_train, y_train = train.data, train.targets
self.assertEqual(x_train.shape[0], 144) # self.assertEqual(x_train.shape[0], 144)
self.assertEqual(y_train.shape[0], 144) # self.assertEqual(y_train.shape[0], 144)
self.assertEqual(x_train.shape[1], 100) # self.assertEqual(x_train.shape[1], 100)
def test_download_test(self): # def test_download_test(self):
rootdir = self.artifacts_dir.rpartition("/")[0] # rootdir = self.artifacts_dir.rpartition("/")[0]
test = pt.datasets.Tecator(root=rootdir, train=False, verbose=False) # test = pt.datasets.Tecator(root=rootdir, train=False, verbose=False)
x_test, y_test = test.data, test.targets # x_test, y_test = test.data, test.targets
self.assertEqual(x_test.shape[0], 71) # self.assertEqual(x_test.shape[0], 71)
self.assertEqual(y_test.shape[0], 71) # self.assertEqual(y_test.shape[0], 71)
self.assertEqual(x_test.shape[1], 100) # self.assertEqual(x_test.shape[1], 100)
def test_class_to_idx(self): # def test_class_to_idx(self):
rootdir = self.artifacts_dir.rpartition("/")[0] # rootdir = self.artifacts_dir.rpartition("/")[0]
test = pt.datasets.Tecator(root=rootdir, train=False, verbose=False) # test = pt.datasets.Tecator(root=rootdir, train=False, verbose=False)
_ = test.class_to_idx # _ = test.class_to_idx
def test_getitem(self): # def test_getitem(self):
rootdir = self.artifacts_dir.rpartition("/")[0] # rootdir = self.artifacts_dir.rpartition("/")[0]
test = pt.datasets.Tecator(root=rootdir, train=False, verbose=False) # test = pt.datasets.Tecator(root=rootdir, train=False, verbose=False)
x, y = test[0] # x, y = test[0]
self.assertEqual(x.shape[0], 100) # self.assertEqual(x.shape[0], 100)
self.assertIsInstance(y, int) # self.assertIsInstance(y, int)
def test_loadable_with_dataloader(self): # def test_loadable_with_dataloader(self):
rootdir = self.artifacts_dir.rpartition("/")[0] # rootdir = self.artifacts_dir.rpartition("/")[0]
test = pt.datasets.Tecator(root=rootdir, train=False, verbose=False) # test = pt.datasets.Tecator(root=rootdir, train=False, verbose=False)
_ = torch.utils.data.DataLoader(test, batch_size=64, shuffle=True) # _ = torch.utils.data.DataLoader(test, batch_size=64, shuffle=True)
def tearDown(self): # def tearDown(self):
self._remove_artifacts() # self._remove_artifacts()