diff options
author | Martin Matusiak <numerodix@gmail.com> | 2021-05-05 12:35:58 +1000 |
---|---|---|
committer | Martin Matusiak <numerodix@gmail.com> | 2021-05-05 12:35:58 +1000 |
commit | 5a27bea4e4e9db63d3b769bf9164e081a1302628 (patch) | |
tree | ab9438ba8c04a5bcceba52871616419dbd814462 | |
parent | 6fba7358f8fef9f48ff9c0936aa706c86cb6bd84 (diff) | |
download | ansicolor-5a27bea4e4e9db63d3b769bf9164e081a1302628.tar.gz |
add tests for highlight_string
-rw-r--r-- | tests/test_colors.py | 52 |
1 files changed, 42 insertions, 10 deletions
diff --git a/tests/test_colors.py b/tests/test_colors.py index 33a7adf..d502057 100644 --- a/tests/test_colors.py +++ b/tests/test_colors.py @@ -107,6 +107,48 @@ def test_get_hightlighter(): assert Colors.Yellow == get_highlighter(1) +def test_highlight_string_one_layer(): + text = "aaabbbaaa" + spanlists = [ + [(3, 6)], + ] + + assert ( + 'aaa' + + get_code(Colors.Green) + + 'bbb' + + get_code(None) + + 'aaa' + ) == highlight_string(text, *spanlists) + +def test_highlight_string_one_color_chosen(): + text = "aaabbbaaa" + spanlists = [ + [(3, 6)], + ] + + assert ( + 'aaa' + + get_code(Colors.Cyan) + + 'bbb' + + get_code(None) + + 'aaa' + ) == highlight_string(text, *spanlists, color=Colors.Cyan) + +def test_highlight_string_nocolor(): + text = "aaabbbaaa" + spanlists = [ + [(3, 6)], + ] + + assert ( + 'aaa' + + get_code(None) + + 'bbb' + + get_code(None) + + 'aaa' + ) == highlight_string(text, *spanlists, nocolor=True) + def test_highlight_string_four_layers(): text = "aaabbbcccdddeeefffeeedddcccbbbaaa" spanlists = [ @@ -142,16 +184,6 @@ def test_highlight_string_four_layers(): + 'aaa' ) == highlight_string(text, *spanlists, colors=colors) -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 ( |