From fe9311e0832b42bb15a79aa280cbe3d919fc9f94 Mon Sep 17 00:00:00 2001 From: "ogueler@idiap.ch" <ogueler@vws110.idiap.ch> Date: Tue, 21 Mar 2023 18:00:28 +0100 Subject: [PATCH] relaxed saving checkpoint occurence assert --- tests/test_cli.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/test_cli.py b/tests/test_cli.py index b1b5c290..b7717459 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -302,6 +302,11 @@ def test_train_pasa_montgomery_from_checkpoint(temporary_basedir): f"instead of the expected {v}:\nOutput:\n{logging_output}" ) + extra_keyword = "Saving checkpoint" + assert extra_keyword in logging_output, ( + f"String '{extra_keyword}' did not appear at least once in the output:\nOutput:\n{logging_output}" + ) + @pytest.mark.skip_if_rc_var_not_set("datadir.montgomery") def test_predict_pasa_montgomery(temporary_basedir, datadir): -- GitLab