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())