mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-05 17:00:58 +01:00
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Tests the xonsh lexer."""
|
|
from __future__ import unicode_literals, print_function
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from xonsh.wizard import (Node, Wizard, Pass, PrettyFormatter,
|
|
Message, Question, StateVisitor)
|
|
|
|
|
|
TREE0 = Wizard(children=[Pass(), Message(message='yo')])
|
|
TREE1 = Question('wakka?', {'jawaka': Pass()})
|
|
|
|
def test_pretty_format_tree0():
|
|
exp = ('Wizard(children=[\n'
|
|
' Pass(),\n'
|
|
" Message('yo')\n"
|
|
'])')
|
|
obs = PrettyFormatter(TREE0).visit()
|
|
assert exp == obs
|
|
assert exp == str(TREE0)
|
|
assert exp.replace('\n', '') == repr(TREE0)
|
|
|
|
|
|
def test_pretty_format_tree1():
|
|
exp = ('Question(\n'
|
|
" question='wakka?',\n"
|
|
' responses={\n'
|
|
" 'jawaka': Pass()\n"
|
|
' }\n'
|
|
')')
|
|
obs = PrettyFormatter(TREE1).visit()
|
|
assert exp == obs
|
|
assert exp == str(TREE1)
|
|
assert exp.replace('\n', '') == repr(TREE1)
|
|
|
|
|
|
def test_state_visitor_store():
|
|
exp = {'rick': [{}, {}, {'and': 'morty'}]}
|
|
sv = StateVisitor()
|
|
sv.store('/rick/2/and', 'morty')
|
|
obs = sv.state
|
|
assert exp == obs
|
|
|
|
exp['rick'][1]['mr'] = 'meeseeks'
|
|
sv.store('/rick/-2/mr', 'meeseeks')
|
|
assert exp == obs
|
|
|
|
flat_exp = {'/': {'rick': [{}, {'mr': 'meeseeks'}, {'and': 'morty'}]},
|
|
'/rick/': [{}, {'mr': 'meeseeks'}, {'and': 'morty'}],
|
|
'/rick/0/': {},
|
|
'/rick/1/': {'mr': 'meeseeks'},
|
|
'/rick/1/mr': 'meeseeks',
|
|
'/rick/2/': {'and': 'morty'},
|
|
'/rick/2/and': 'morty',
|
|
}
|
|
flat_obs = sv.flatten()
|
|
assert flat_exp == flat_obs
|