diff --git a/tests/test_commands.py b/tests/test_commands.py index 110964a..3a6be40 100644 --- a/tests/test_commands.py +++ b/tests/test_commands.py @@ -64,7 +64,11 @@ def test_basic_set_flag_format_original_option_cli(): # We pass -o SKY-1111-000 and input "2" to choose the second suggested pattern result = runner.invoke(basic_group, ["set-flag-format", "-o", "SKY-1111-000"], input="2\n") assert result.exit_code == 0 - assert "Suggested regex patterns:" in result.output + comp_name = original_config.get("Competition", {}).get("competition", "") + if comp_name: + assert f"Suggested regex patterns for competition {comp_name}:" in result.output + else: + assert "Suggested regex patterns:" in result.output assert "Select a pattern index" in result.output assert "Flag format set to" in result.output