def test_save_wrong(self, config): X_train_set, y_train_set, X_test, y_test, stats = MainTransformer.get_training_and_test_set(dataset, 'Pollutant', 'Uncertainty', size=0.5, normalize=True) gp = GaussianProcesses() gp.train(X_train_set, y_train_set, stats=stats) result, msg = gp.save_model(config) assert not result and isinstance(msg, str)
def test_save(self, config): X_train_set, y_train_set, X_test, y_test, stats = MainTransformer.get_training_and_test_set(dataset, 'Pollutant', 'Uncertainty', size=0.5, normalize=True) gp = GaussianProcesses() gp.train(X_train_set, y_train_set, stats=stats) result, msg = gp.save_model(config) global full_gp full_gp = gp assert result and msg is None