From b6dfdaff33f19757b1cb9b3bf1d17f28b94859d4 Mon Sep 17 00:00:00 2001 From: David Smith Date: Tue, 28 Jul 2020 21:24:09 +0100 Subject: [PATCH] Completed test coverage for colorize(). --- tests/utils_tests/test_termcolors.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/tests/utils_tests/test_termcolors.py b/tests/utils_tests/test_termcolors.py index 0f517d1391..0469137a9c 100644 --- a/tests/utils_tests/test_termcolors.py +++ b/tests/utils_tests/test_termcolors.py @@ -183,3 +183,27 @@ class TermColorTests(unittest.TestCase): self.assertEqual(colorize(text=None, opts=('noreset',)), '\x1b[m') self.assertEqual(colorize(text='', opts=('noreset',)), '\x1b[m') + + def test_colorize_reset(self): + self.assertEqual(colorize(text='', opts=('reset',)), '\x1b[0m') + + def test_colorize_fg_bg(self): + self.assertEqual(colorize(text='Test', fg='red'), '\x1b[31mTest\x1b[0m') + self.assertEqual(colorize(text='Test', bg='red'), '\x1b[41mTest\x1b[0m') + # Ignored kwarg. + self.assertEqual(colorize(text='Test', other='red'), '\x1b[mTest\x1b[0m') + + def test_colorize_opts(self): + self.assertEqual( + colorize(text='Test', opts=('bold', 'underscore')), + '\x1b[1;4mTest\x1b[0m', + ) + self.assertEqual( + colorize(text='Test', opts=('blink',)), + '\x1b[5mTest\x1b[0m', + ) + # Ignored opts. + self.assertEqual( + colorize(text='Test', opts=('not_an_option',)), + '\x1b[mTest\x1b[0m', + )