fix: improved test flakiness

ErikBjare · Aug 20, 2024 · 0758c4d · 0758c4d
1 parent 79355bc
commit 0758c4d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/test_cli.py b/tests/test_cli.py
@@ -290,8 +290,13 @@ def test_subagent(args: list[str], runner: CliRunner):
 
     # apparently this is not obviously 610
     accepteds = ["377", "610"]
-    assert any(accepted in result.output for accepted in accepteds)
-    assert any(accepted in result.output.split("```")[-1] for accepted in accepteds)
+    assert any([accepted in result.output for accepted in accepteds])
+    assert any(
+        [
+            accepted in "```".join(result.output.split("```")[-2:])
+            for accepted in accepteds
+        ]
+    )
 
 
 @pytest.mark.slow