Updated center_text to support ansi escape sequences and characters with display widths greater than 1.

Also added left and right justification functions.

1 files changed, 11 insertions, 14 deletions

diff --git a/tests/test_utils.py b/tests/test_utils.py
index e4b9169c..2c43371f 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -293,24 +293,21 @@ def test_context_flag_exit_err(context_flag):
         context_flag.__exit__()
 
 
-def test_center_text_pad_none():
+def test_center_text_pad_equals():
     msg = 'foo'
-    centered = cu.center_text(msg, pad=None)
-    expected_center = ' ' + msg + ' '
-    assert expected_center in centered
+    fill_char = '='
+    centered = cu.center_text(msg, fill_char=fill_char)
+    assert msg in centered
+    assert centered.startswith(fill_char)
+    assert centered.endswith(fill_char)
     letters_in_centered = set(centered)
     letters_in_msg = set(msg)
     assert len(letters_in_centered) == len(letters_in_msg) + 1
 
-def test_center_text_pad_equals():
+
+def test_center_text_pad_blank():
     msg = 'foo'
-    pad = '='
-    centered = cu.center_text(msg, pad=pad)
-    expected_center = ' ' + msg + ' '
-    assert expected_center in centered
-    assert centered.startswith(pad)
-    assert centered.endswith(pad)
-    letters_in_centered = set(centered)
-    letters_in_msg = set(msg)
-    assert len(letters_in_centered) == len(letters_in_msg) + 2
+    fill_char = ''
 
+    with pytest.raises(ValueError):
+        cu.center_text(msg, fill_char=fill_char)