summaryrefslogtreecommitdiff
path: root/tests/test_utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_utils.py')
-rw-r--r--tests/test_utils.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/test_utils.py b/tests/test_utils.py
index 88eb0901..6e3600db 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -29,18 +29,18 @@ def test_ansi_safe_wcswidth():
def test_style():
base_str = HELLO_WORLD
ansi_str = Fore.BLUE + Back.GREEN + base_str + Fore.RESET + Back.RESET
- assert ansi.style(base_str, fg='blue', bg='green') == ansi_str
+ assert ansi.style(base_str, ansi.TextStyle(fg='blue', bg='green')) == ansi_str
def test_style_color_not_exist():
base_str = HELLO_WORLD
try:
- ansi.style(base_str, fg='hello', bg='green')
+ ansi.style(base_str, ansi.TextStyle(fg='fake', bg='green'))
assert False
except ValueError:
assert True
try:
- ansi.style(base_str, fg='blue', bg='hello')
+ ansi.style(base_str, ansi.TextStyle(fg='blue', bg='fake'))
assert False
except ValueError:
assert True