Adding upstream version 3.0.36.upstream/3.0.36upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 18 insertions, 0 deletions

diff --git a/tests/test_vt100_output.py b/tests/test_vt100_output.py
new file mode 100644
index 0000000..f7163da
--- /dev/null
+++ b/tests/test_vt100_output.py
@@ -0,0 +1,18 @@
+from prompt_toolkit.output.vt100 import _get_closest_ansi_color
+
+
+def test_get_closest_ansi_color():
+    # White
+    assert _get_closest_ansi_color(255, 255, 255) == "ansiwhite"
+    assert _get_closest_ansi_color(250, 250, 250) == "ansiwhite"
+
+    # Black
+    assert _get_closest_ansi_color(0, 0, 0) == "ansiblack"
+    assert _get_closest_ansi_color(5, 5, 5) == "ansiblack"
+
+    # Green
+    assert _get_closest_ansi_color(0, 255, 0) == "ansibrightgreen"
+    assert _get_closest_ansi_color(10, 255, 0) == "ansibrightgreen"
+    assert _get_closest_ansi_color(0, 255, 10) == "ansibrightgreen"
+
+    assert _get_closest_ansi_color(220, 220, 100) == "ansiyellow"