diff --git a/tests/test_term_utils.py b/tests/test_term_utils.py index fccb440..b49ab1d 100644 --- a/tests/test_term_utils.py +++ b/tests/test_term_utils.py @@ -44,19 +44,20 @@ def test_parse_with_escape_blocks(): 'this is a prompt {start}my first context block{end} and then ```my second context block```', {'bar': '1.0', 'baz': '2.0'}, ), + ( + "{start}this is a prompt --bar=1.0 --baz=2.0{end}", + "{start}this is a prompt --bar=1.0 --baz=2.0{end}", + {} + ) ] delimiters = ["```", '"""', "`"] for start, end in [(d, d) for d in delimiters]: - assert parse_args(f"{start}this is a prompt --bar=1.0 --baz=2.0{end}") == ( - f"{start}this is a prompt --bar=1.0 --baz=2.0{end}", - {}, - ) for prompt, expected_prompt, expected_args in test_cases: formatted_prompt = prompt.format(start=start, end=end) formatted_expected_prompt = expected_prompt.format(start=start, end=end) assert parse_args(formatted_prompt) == ( formatted_expected_prompt, expected_args, - ) + ) \ No newline at end of file