mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-05 17:00:58 +01:00
24 lines
804 B
Python
24 lines
804 B
Python
"""Tests some tools function for prompt_toolkit integration."""
|
|
from __future__ import unicode_literals, print_function
|
|
|
|
import nose
|
|
from nose.tools import assert_equal
|
|
|
|
from xonsh.tools import format_prompt_for_prompt_toolkit
|
|
from xonsh.tools import TERM_COLORS
|
|
from xonsh.environ import format_prompt
|
|
|
|
|
|
|
|
def test_format_prompt_for_prompt_toolkit():
|
|
templ = ('>>> {BOLD_BLUE}~/xonsh {WHITE} (main){NO_COLOR}')
|
|
prompt = format_prompt(templ, TERM_COLORS)
|
|
token_names, color_styles, strings = format_prompt_for_prompt_toolkit(prompt)
|
|
assert_equal(token_names, ['NO_COLOR', 'BOLD_BLUE', 'WHITE', 'NO_COLOR'])
|
|
assert_equal(color_styles, ['', 'bold #0000FF', '#FFFFFF', ''])
|
|
assert_equal(strings, ['>>> ', '~/xonsh ', ' (main)', ''])
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
nose.runmodule()
|