def test_pformat_py38_type_comments(tmpdir, capsys):
expected = '''\
Module(
body=[
FunctionDef(
lineno=1,
col_offset=0,
end_lineno=2,
end_col_offset=8,
name='f',
args=arguments(posonlyargs=[], args=[], vararg=None, kwonlyargs=[], kw_defaults=[], kwarg=None, defaults=[]),
body=[Pass(lineno=2, col_offset=4, end_lineno=2, end_col_offset=8)],
decorator_list=[],
returns=None,
type_comment='() -> None',
),
],
type_ignores=[TypeIgnore(lineno=2, tag='')],
)
''' # noqa: E501
mod = ('def f(): # type: () -> None\n' ' pass # type: ignore\n')
f = tmpdir.join('test.py')
f.write(mod)
astpretty.main((f.strpath, ))
out, _ = capsys.readouterr()
assert out == expected
def test_main_hide_offsets(capsys, tmpdir):
expected_38 = '''\
Module(
body=[
Assign(
targets=[Name(id='x', ctx=Store())],
value=Name(id='y', ctx=Load()),
type_comment=None,
),
],
type_ignores=[],
)
'''
expected_lt38 = '''\
Module(
body=[
Assign(
targets=[Name(id='x', ctx=Store())],
value=Name(id='y', ctx=Load()),
),
],
)
'''
expected = expected_38 if sys.version_info >= (3, 8) else expected_lt38
f = tmpdir.join('test.py')
f.write('x = y\n')
astpretty.main((f.strpath, '--no-show-offsets'))
out, _ = capsys.readouterr()
assert out == expected
def test_main_with_offsets(capsys, tmpdir):
expected_38 = '''\
Module(
body=[
Assign(
lineno=1,
col_offset=0,
end_lineno=1,
end_col_offset=5,
targets=[Name(lineno=1, col_offset=0, end_lineno=1, end_col_offset=1, id='x', ctx=Store())],
value=Name(lineno=1, col_offset=4, end_lineno=1, end_col_offset=5, id='y', ctx=Load()),
type_comment=None,
),
],
type_ignores=[],
)
''' # noqa: E501
expected_lt38 = '''\
Module(
body=[
Assign(
lineno=1,
col_offset=0,
targets=[Name(lineno=1, col_offset=0, id='x', ctx=Store())],
value=Name(lineno=1, col_offset=4, id='y', ctx=Load()),
),
],
)
'''
expected = expected_38 if sys.version_info >= (3, 8) else expected_lt38
f = tmpdir.join('test.py')
f.write('x = y\n')
astpretty.main((f.strpath, ))
out, _ = capsys.readouterr()
assert out == expected
def test_typedast_support_cmdline_3(tmpdir, capsys): # pragma: no cover
f = tmpdir.join('f.py')
f.write(TYPED_SRC)
assert not astpretty.main((str(f), '--typed-3'))
out, _ = capsys.readouterr()
assert out == TYPED3_OUT
def test_main_hide_offsets(capsys, tmpdir):
f = tmpdir.join('test.py')
f.write('x = 5\n')
astpretty.main((f.strpath, '--no-show-offsets'))
out, _ = capsys.readouterr()
assert out == '''\
def test_main(capsys, tmpdir):
f = tmpdir.join('test.py')
f.write('x = 5\n')
astpretty.main((f.strpath, ))
out, _ = capsys.readouterr()
assert out == '''\
