Tune test to search param values only in the intended line

NatLibFi · Jul 5, 2019 · f4fc1d0 · f4fc1d0
1 parent 53bf269
commit f4fc1d0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -159,9 +159,11 @@ def test_train_fasttext_params():
          '--backend-param', 'fasttext.epoch=0',
          '-v', 'DEBUG'])
     assert not result.exception
-    assert "'dim': 1" in result.output
-    assert "'lr': 42.1" in result.output
-    assert "'epoch': 0" in result.output
+    for line in result.output.split("\n"):
+        if 'Backend fasttext: Model parameters:' in line:
+            assert "'dim': 1" in line
+            assert "'lr': 42.1" in line
+            assert "'epoch': 0" in line
     assert result.exit_code == 0