From bced087d58ec5a7e0ad5e8d7c3af6e6e84df56ef Mon Sep 17 00:00:00 2001 From: Kevin Van Brunt Date: Thu, 15 Nov 2018 15:12:01 -0500 Subject: Added wrapper for wcswidth that removes ansi escape characters --- tests/test_utils.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'tests/test_utils.py') diff --git a/tests/test_utils.py b/tests/test_utils.py index 807bc0fd..2f4723e4 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -21,6 +21,11 @@ def test_strip_ansi(): assert base_str != ansi_str assert base_str == cu.strip_ansi(ansi_str) +def test_display_width(): + base_str = HELLO_WORLD + ansi_str = Fore.GREEN + base_str + Fore.RESET + assert cu.display_width(ansi_str) != len(ansi_str) + def test_strip_quotes_no_quotes(): base_str = HELLO_WORLD stripped = cu.strip_quotes(base_str) -- cgit v1.2.1