def test_init_training(self):
"""
Tests initialization of DataManager
:return:
"""
dataset = Dataset(self.training_directory)
self.assertIsInstance(dataset, Dataset)
self.assertTrue(dataset.is_training())
def test_init_prediction(self):
"""
Tests initialization of DataManager
:return:
"""
dataset = Dataset(self.prediction_directory)
self.assertIsInstance(dataset, Dataset)
self.assertFalse(dataset.is_training())
