intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 18 insertions, 0 deletions

diff --git a/contrib/python/prompt-toolkit/py3/tests/test_vt100_output.py b/contrib/python/prompt-toolkit/py3/tests/test_vt100_output.py
new file mode 100644
index 00000000000..f7163da8393
--- /dev/null
+++ b/contrib/python/prompt-toolkit/py3/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"