mirror of
https://github.com/xonsh/xonsh.git
synced 2025-03-05 17:00:58 +01:00
33 lines
788 B
Python
33 lines
788 B
Python
"""Xonsh AST tests."""
|
|
from xonsh import ast
|
|
from xonsh.ast import Tuple, Name, Store, min_line
|
|
|
|
import pytest
|
|
|
|
from tools import check_parse
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def xonsh_execer_autouse(xonsh_execer):
|
|
return xonsh_execer
|
|
|
|
def test_gather_names_name():
|
|
node = Name(id='y', ctx=Store())
|
|
exp = {'y'}
|
|
obs = ast.gather_names(node)
|
|
assert exp == obs
|
|
|
|
|
|
def test_gather_names_tuple():
|
|
node = Tuple(elts=[Name(id='y', ctx=Store()),
|
|
Name(id='z', ctx=Store())])
|
|
exp = {'y', 'z'}
|
|
obs = ast.gather_names(node)
|
|
assert exp == obs
|
|
|
|
def test_multilline_num():
|
|
code = ('x = 1\n'
|
|
'ls -l\n') # this second line wil be transformed
|
|
tree = check_parse(code)
|
|
lsnode = tree.body[1]
|
|
assert 2 == min_line(lsnode)
|