diff --git a/tests/test_cli/test_cli.py b/tests/test_cli/test_cli.py index de08919..b4f14a9 100644 --- a/tests/test_cli/test_cli.py +++ b/tests/test_cli/test_cli.py @@ -230,7 +230,7 @@ def test_new_create_scoped_page_mismatched_token( config_scope = _new_get_config_scope("some_scope") inputs = ("some_input",) - if not example: + if not example[0]: config_scope["example"] = None # act @@ -239,7 +239,13 @@ def test_new_create_scoped_page_mismatched_token( # assert assert system_exit.value.code == 1 - assert ( - capsys.readouterr().out - == f'[TEMBO] Your tembo config.yml/template specifies 0 input tokens, you gave 1. Example: tembo new some_scope 🐘\n' - ) + if not example[0]: + assert ( + capsys.readouterr().out + == f'[TEMBO] Your tembo config.yml/template specifies 0 input tokens, you gave 1 🐘\n' + ) + else: + assert ( + capsys.readouterr().out + == f'[TEMBO] Your tembo config.yml/template specifies 0 input tokens, you gave 1. Example: tembo new some_scope 🐘\n' + )