Commit 95b71422 authored by lukas leufen's avatar lukas leufen

test for save model was still erroneous

parent bc989d62
Pipeline #39302 passed with stages
in 7 minutes and 31 seconds
......@@ -205,14 +205,14 @@ class TestTraining:
assert ready_to_train.model.history.epoch == [0, 1]
assert len(glob.glob(os.path.join(path, "plots", "TestExperiment_history_*.pdf"))) == 2
def test_save_model(self, init_without_run, path, caplog):
def test_save_model(self, init_without_run, model_path, caplog):
caplog.set_level(logging.DEBUG)
model_name = "test_model.h5"
assert model_name not in os.listdir(path)
assert model_name not in os.listdir(model_path)
init_without_run.save_model()
message = PyTestRegex(f"save best model to {os.path.join(path, model_name)}")
message = PyTestRegex(f"save best model to {os.path.join(model_path, model_name)}")
assert caplog.record_tuples[1] == ("root", 10, message)
assert model_name in os.listdir(path)
assert model_name in os.listdir(model_path)
def test_load_best_model_no_weights(self, init_without_run, caplog):
caplog.set_level(logging.DEBUG)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment