From a9ce49bfc46610e03d62a1e4afbfdde9f5b0f7ee Mon Sep 17 00:00:00 2001 From: Todd Leonhardt Date: Sun, 16 Jun 2019 14:41:49 -0400 Subject: Added center_text() utility function for centering text in a terminal along with unit tests for it This function isn't used yet, but I have plans to use it when improving the output of transcript testing --- tests/test_utils.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) (limited to 'tests/test_utils.py') 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 + -- cgit v1.2.1