diff options
Diffstat (limited to 'tests/test_cmd2.py')
| -rw-r--r-- | tests/test_cmd2.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/test_cmd2.py b/tests/test_cmd2.py index f4c8d3e7..2c6a4eb0 100644 --- a/tests/test_cmd2.py +++ b/tests/test_cmd2.py @@ -1447,20 +1447,20 @@ def test_poutput_empty_string(outsim_app): msg = '' outsim_app.poutput(msg) out = outsim_app.stdout.getvalue() - expected = msg + expected = '\n' assert out == expected def test_poutput_none(outsim_app): msg = None outsim_app.poutput(msg) out = outsim_app.stdout.getvalue() - expected = '' + expected = 'None\n' assert out == expected def test_poutput_color_always(outsim_app): msg = 'Hello World' outsim_app.colors = 'Always' - outsim_app.poutput(msg, fg='cyan') + outsim_app.poutput(utils.style_message(msg, fg='cyan')) out = outsim_app.stdout.getvalue() expected = Fore.CYAN + msg + Fore.RESET + '\n' assert out == expected @@ -1468,7 +1468,7 @@ def test_poutput_color_always(outsim_app): def test_poutput_color_never(outsim_app): msg = 'Hello World' outsim_app.colors = 'Never' - outsim_app.poutput(msg, fg='cyan') + outsim_app.poutput(utils.style_message(msg, fg='cyan')) out = outsim_app.stdout.getvalue() expected = msg + '\n' assert out == expected |