def test_should_get_dataset(self):
data = DatasetManager("./tests/resources/local_data")
dataset = {
"local_test": {
"source": "./tests/resources/local_data/train.csv",
"description": "my little dataset local"
}
}
self.assertDictEqual(data.get_dataset("local_test"),
dataset.get("local_test"))
def test_should_get_dataset_unknown(self):
data = DatasetManager("./tests/resources/local_data")
with self.assertRaises(IOError):
data.get_dataset("unknown_test")
