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