Merge pull request #703 from python-cmd2/public_api

Minimize public API of cmd2.Cmd class

1 files changed, 23 insertions, 0 deletions

diff --git a/tests/test_utils.py b/tests/test_utils.py
index b43eb10c..44421b93 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -276,3 +276,26 @@ def test_context_flag_bool(context_flag):
 def test_context_flag_exit_err(context_flag):
     with pytest.raises(ValueError):
         context_flag.__exit__()
+
+
+def test_center_text_pad_none():
+    msg = 'foo'
+    centered = cu.center_text(msg, pad=None)
+    expected_center = ' ' + msg + ' '
+    assert expected_center in centered
+    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():
+    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
+