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