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 == '''\