def add_experiment(name: Text, type: Text, train_data_filename: Text, test_data_filename: Text, train_data, test_data): experiment = Experiment(name=name, type=type) db.session.add(experiment) db.session.commit() experiment_locator = Locator(experiment.id, train_data_filename=train_data_filename, test_data_filename=test_data_filename) with open( os.path.join(experiment_locator.get_train_data_dir(), train_data_filename), 'wb') as copied_train_data: for line in train_data.readlines(): copied_train_data.write(line) with open( os.path.join(experiment_locator.get_test_data_dir(), test_data_filename), 'wb') as copied_test_data: for line in test_data.readlines(): copied_test_data.write(line) experiment.train_data = train_data_filename experiment.test_data = test_data_filename db.session.commit() return experiment
def test_get_train_data_dir(self): """Ensure get_train_data_dir behaves properly""" locator = Locator(id=1, train_data_filename='pima_indians.csv', test_data_filename='pima_indians.csv') self.assertIn('/experiments/1/data/train', locator.get_train_data_dir())