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.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/test_utils.py b/tests/test_utils.py
index f43aed4e..ada8bc86 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -27,8 +27,8 @@ def test_ansi_safe_wcswidth():
def test_style_message():
base_str = HELLO_WORLD
- ansi_str = Fore.BLUE + Back.GREEN + base_str + Fore.RESET + Back.RESET + '\n\n'
- assert cu.style_message(base_str, end='\n\n', fg='blue', bg='green')
+ ansi_str = Fore.BLUE + Back.GREEN + base_str + Fore.RESET + Back.RESET
+ assert cu.style_message(base_str, fg='blue', bg='green') == ansi_str
def test_style_message_color_not_exist():
base_str = HELLO_WORLD