1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
from baron import parse, dumps, tokenize
def test_regression_trailing_comment_after_colon():
assert parse("def a(): # pouf\n pass")
def test_regression_trailing_comment_after_colon_no_space():
assert parse("def a():# pouf\n pass")
def test_regression_trailing_comment_after_colon_dump():
code = "def a(): # pouf\n pass\n"
assert dumps(parse(code)) == code
def test_regression_trailing_comment_after_colon_no_space_dump():
code = "def a():# pouf\n pass\n"
assert dumps(parse(code)) == code
def test_comment_in_middle_of_ifelseblock():
code = 'if a:\n pass\n# comment\nelse:\n pass\n'
assert dumps(parse(code)) == code
def test_new_float_notation():
code = 'a._'
assert tokenize(code)[:-1] == [('NAME', 'a'), ('DOT', '.'), ('NAME', '_')]
|
