diff options
author | Martin Matusiak <numerodix@gmail.com> | 2021-05-05 12:05:47 +1000 |
---|---|---|
committer | Martin Matusiak <numerodix@gmail.com> | 2021-05-05 12:05:47 +1000 |
commit | 45637c18dd67a44a9ce82f22064ed4b3657f330c (patch) | |
tree | f71a5cf3ebc28e6eeb0e01208b04f020a8a6e977 | |
parent | a2a85c08d43889c01528f9ca824bd2692f0304dc (diff) | |
download | ansicolor-45637c18dd67a44a9ce82f22064ed4b3657f330c.tar.gz |
add test for highlight_string
-rw-r--r-- | tests/test_colors.py | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/tests/test_colors.py b/tests/test_colors.py index acc7375..901f669 100644 --- a/tests/test_colors.py +++ b/tests/test_colors.py @@ -12,6 +12,7 @@ from ansicolor import strip_escapes from ansicolor import wrap_string from ansicolor import get_code_v2 from ansicolor import colorize_v2 +from ansicolor import highlight_string from ansicolor import set_term_title from ansicolor import write_out from ansicolor import write_err @@ -100,12 +101,23 @@ def test_coloring(): assert '\033[0;0;31m' + 'hi' + '\033[0;0m' == red('hi') -def test_highlights(): +def test_get_hightlighter(): # can I get a highlighter? assert Colors.Green == get_highlighter(0) assert Colors.Yellow == get_highlighter(1) +def test_highlight_string__more_spans_than_colors(): + first_highlighter = get_highlighter(0) + assert ( + 'aaa' + + get_code(first_highlighter) + + 'bbb' + + get_code(None) + + 'aaa' + ) == highlight_string("aaabbbaaa", [(3,6)], colors=[]) + + def test_colorize(): assert ( get_code(Colors.Red) |