AutoGPT

1
import pytest
2

3
from autogpt.logs.utils import remove_color_codes
4

5

6
@pytest.mark.parametrize(
7
    "raw_text, clean_text",
8
    [
9
        (
10
            "COMMAND = \x1b[36mbrowse_website\x1b[0m  "
11
            "ARGUMENTS = \x1b[36m{'url': 'https://www.google.com',"
12
            " 'question': 'What is the capital of France?'}\x1b[0m",
13
            "COMMAND = browse_website  "
14
            "ARGUMENTS = {'url': 'https://www.google.com',"
15
            " 'question': 'What is the capital of France?'}",
16
        ),
17
        (
18
            "{'Schaue dir meine Projekte auf github () an, als auch meine Webseiten': "
19
            "'https://github.com/Significant-Gravitas/AutoGPT,"
20
            " https://discord.gg/autogpt und https://twitter.com/Auto_GPT'}",
21
            "{'Schaue dir meine Projekte auf github () an, als auch meine Webseiten': "
22
            "'https://github.com/Significant-Gravitas/AutoGPT,"
23
            " https://discord.gg/autogpt und https://twitter.com/Auto_GPT'}",
24
        ),
25
        ("", ""),
26
        ("hello", "hello"),
27
        ("hello\x1B[31m world", "hello world"),
28
        ("\x1B[36mHello,\x1B[32m World!", "Hello, World!"),
29
        (
30
            "\x1B[1m\x1B[31mError:\x1B[0m\x1B[31m file not found",
31
            "Error: file not found",
32
        ),
33
    ],
34
)
35
def test_remove_color_codes(raw_text, clean_text):
36
    assert remove_color_codes(raw_text) == clean_text
37