add ydb deps

1 files changed, 40 insertions, 0 deletions

diff --git a/contrib/python/prompt-toolkit/py2/tests/test_style.py b/contrib/python/prompt-toolkit/py2/tests/test_style.py
new file mode 100644
index 0000000000..fc839ec95f
--- /dev/null
+++ b/contrib/python/prompt-toolkit/py2/tests/test_style.py
@@ -0,0 +1,40 @@
+from __future__ import unicode_literals
+
+from prompt_toolkit.styles import Attrs, style_from_dict
+from prompt_toolkit.token import Token
+
+
+def test_style_from_dict():
+    style = style_from_dict({
+        Token.A: '#ff0000 bold underline italic',
+        Token.B: 'bg:#00ff00 blink reverse',
+    })
+
+    expected = Attrs(color='ff0000', bgcolor=None, bold=True,
+                     underline=True, italic=True, blink=False, reverse=False)
+    assert style.get_attrs_for_token(Token.A) == expected
+
+    expected = Attrs(color=None, bgcolor='00ff00', bold=False,
+                     underline=False, italic=False, blink=True, reverse=True)
+    assert style.get_attrs_for_token(Token.B) == expected
+
+
+def test_style_inheritance():
+    style = style_from_dict({
+        Token: '#ff0000',
+        Token.A.B.C: 'bold',
+        Token.A.B.C.D: '#ansired',
+        Token.A.B.C.D.E: 'noinherit blink'
+    })
+
+    expected = Attrs(color='ff0000', bgcolor=None, bold=True,
+                     underline=False, italic=False, blink=False, reverse=False)
+    assert style.get_attrs_for_token(Token.A.B.C) == expected
+
+    expected = Attrs(color='ansired', bgcolor=None, bold=True,
+                     underline=False, italic=False, blink=False, reverse=False)
+    assert style.get_attrs_for_token(Token.A.B.C.D) == expected
+
+    expected = Attrs(color=None, bgcolor=None, bold=False,
+                     underline=False, italic=False, blink=True, reverse=False)
+    assert style.get_attrs_for_token(Token.A.B.C.D.E) == expected