diff options
Diffstat (limited to 'autogpts/autogpt/tests/unit/test_logs.py')
-rw-r--r-- | autogpts/autogpt/tests/unit/test_logs.py | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/autogpts/autogpt/tests/unit/test_logs.py b/autogpts/autogpt/tests/unit/test_logs.py new file mode 100644 index 000000000..1ded61f31 --- /dev/null +++ b/autogpts/autogpt/tests/unit/test_logs.py @@ -0,0 +1,36 @@ +import pytest + +from autogpt.logs.utils import remove_color_codes + + +@pytest.mark.parametrize( + "raw_text, clean_text", + [ + ( + "COMMAND = \x1b[36mbrowse_website\x1b[0m " + "ARGUMENTS = \x1b[36m{'url': 'https://www.google.com'," + " 'question': 'What is the capital of France?'}\x1b[0m", + "COMMAND = browse_website " + "ARGUMENTS = {'url': 'https://www.google.com'," + " 'question': 'What is the capital of France?'}", + ), + ( + "{'Schaue dir meine Projekte auf github () an, als auch meine Webseiten': " + "'https://github.com/Significant-Gravitas/AutoGPT," + " https://discord.gg/autogpt und https://twitter.com/Auto_GPT'}", + "{'Schaue dir meine Projekte auf github () an, als auch meine Webseiten': " + "'https://github.com/Significant-Gravitas/AutoGPT," + " https://discord.gg/autogpt und https://twitter.com/Auto_GPT'}", + ), + ("", ""), + ("hello", "hello"), + ("hello\x1B[31m world", "hello world"), + ("\x1B[36mHello,\x1B[32m World!", "Hello, World!"), + ( + "\x1B[1m\x1B[31mError:\x1B[0m\x1B[31m file not found", + "Error: file not found", + ), + ], +) +def test_remove_color_codes(raw_text, clean_text): + assert remove_color_codes(raw_text) == clean_text |