def test_retrieve_nonexistent_file(self):
fake_file = 'nonexistentfile.data'
r = Retriever(fake_file)
with self.assertRaises(errors.RetrieverError):
r.load()
def test_retrieve_brainwave(self):
retriever = Retriever('datasets/brainwave/plrx.txt', target_column=-1)
data, target = retriever.load().split_target().retrieve()
# Assert number of samples.
self.assertEqual(182, len(data))
# Assert number of features.
self.assertEqual(12, len(data[1]))
# Assert number of samples in target array.
self.assertEqual(182, len(target))