def test_gather_load_store_names_tuple():
node = Tuple(elts=[Name(id="y", ctx=Store()), Name(id="z", ctx=Store())])
lexp = set()
sexp = {"y", "z"}
lobs, sobs = ast.gather_load_store_names(node)
assert lexp == lobs
assert sexp == sobs
def test_add_inline_comment_class():
code = """
# comment 1
class Class: # comment 2
# comment 3
'''
# not a comment
'''
# comment 4
x = True # comment 5
# comment 6
# comment 7
"""
exp = Module(
body=[
Comment(s="# comment 1", lineno=1, col_offset=0),
NodeWithComment(
node=ClassDef(
name="Class",
bases=[],
keywords=[],
body=[
Comment(s="# comment 3", lineno=3, col_offset=4),
Expr(
value=Str(
s="\n # not a comment\n ", lineno=4, col_offset=4
),
lineno=4,
col_offset=4,
),
Comment(s="# comment 4", lineno=8, col_offset=4),
NodeWithComment(
node=Assign(
targets=[
Name(id="x", ctx=Store(), lineno=9, col_offset=4)
],
value=NameConstant(value=True, lineno=9, col_offset=8),
lineno=9,
col_offset=4,
),
comment=Comment(s="# comment 5", lineno=9, col_offset=14),
lineno=9,
col_offset=4,
),
Comment(s="# comment 6", lineno=10, col_offset=4),
],
decorator_list=[],
lineno=2,
col_offset=0,
),
comment=Comment(s="# comment 2", lineno=2, col_offset=14),
lineno=2,
col_offset=0,
),
Comment(s="# comment 7", lineno=12, col_offset=0),
]
)
check_add_comments(code, exp)
def test_add_inline_comment_if_else():
code = """
# comment 1
if True: # comment 2
# comment 3
x = 1 # comment 4
# comment 5
else: # comment 6
# comment 7
x = 4 # comment 8
# comment 9
# comment 10
"""
exp = Module(
body=[
Comment(s="# comment 1"),
IfWithComments(
node=If(
test=NameConstant(value=True),
body=[
Comment(s="# comment 3"),
NodeWithComment(
node=Assign(
targets=[Name(id="x", ctx=Store())], value=Num(n=1)
),
comment=Comment(s="# comment 4"),
),
Comment(s="# comment 5"),
],
orelse=[
Comment(s="# comment 7"),
NodeWithComment(
node=Assign(
targets=[Name(id="x", ctx=Store())], value=Num(n=4)
),
comment=Comment(s="# comment 8"),
),
Comment(s="# comment 9"),
],
),
comment=Comment(s="# comment 2"),
elsecomment=Comment(s="# comment 6"),
),
Comment(s="# comment 10"),
]
)
check_add_comments(code, exp)
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_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_add_inline_comment_elif():
code = """
# comment 1
if True: # comment 2
# comment 3
x = 1 # comment 4
# comment 5
elif True: # comment 6
# comment 7
x = 2 # comment 8
# comment 9
elif False: # comment 10
# comment 11
x = 3 # comment 12
# comment 13
else: # comment 14
# comment 15
x = 4 # comment 16
# comment 17
# comment 18
"""
exp = Module(
body=[
Comment(s="# comment 1", lineno=1, col_offset=0),
IfWithComments(
node=If(
test=NameConstant(value=True, lineno=2, col_offset=3),
body=[
Comment(s="# comment 3", lineno=3, col_offset=4),
NodeWithComment(
node=Assign(
targets=[
Name(id="x", ctx=Store(), lineno=4, col_offset=4)
],
value=Num(n=1, lineno=4, col_offset=8),
lineno=4,
col_offset=4,
),
comment=Comment(s="# comment 4", lineno=4, col_offset=11),
lineno=4,
col_offset=4,
),
Comment(s="# comment 5", lineno=5, col_offset=4),
],
orelse=[
IfWithComments(
node=If(
test=NameConstant(value=True, lineno=6, col_offset=5),
body=[
Comment(s="# comment 7", lineno=7, col_offset=4),
NodeWithComment(
node=Assign(
targets=[
Name(
id="x",
ctx=Store(),
lineno=8,
col_offset=4,
)
],
value=Num(n=2, lineno=8, col_offset=8),
lineno=8,
col_offset=4,
),
comment=Comment(
s="# comment 8", lineno=8, col_offset=11
),
lineno=8,
col_offset=4,
),
Comment(s="# comment 9", lineno=9, col_offset=4),
],
orelse=[
IfWithComments(
node=If(
test=NameConstant(
value=False, lineno=10, col_offset=5
),
body=[
Comment(
s="# comment 11",
lineno=11,
col_offset=4,
),
NodeWithComment(
node=Assign(
targets=[
Name(
id="x",
ctx=Store(),
lineno=12,
col_offset=4,
)
],
value=Num(
n=3, lineno=12, col_offset=8
),
lineno=12,
col_offset=4,
),
comment=Comment(
s="# comment 12",
lineno=12,
col_offset=11,
),
lineno=12,
col_offset=4,
),
Comment(
s="# comment 13",
lineno=13,
col_offset=4,
),
],
orelse=[
Comment(
s="# comment 15",
lineno=15,
col_offset=4,
),
NodeWithComment(
node=Assign(
targets=[
Name(
id="x",
ctx=Store(),
lineno=16,
col_offset=4,
)
],
value=Num(
n=4, lineno=16, col_offset=8
),
lineno=16,
col_offset=4,
),
comment=Comment(
s="# comment 16",
lineno=16,
col_offset=11,
),
lineno=16,
col_offset=4,
),
Comment(
s="# comment 17",
lineno=17,
col_offset=4,
),
],
lineno=10,
col_offset=5,
),
comment=Comment(
s="# comment 10", lineno=10, col_offset=12
),
elsecomment=Comment(
s="# comment 14", lineno=14, col_offset=6
),
lineno=10,
col_offset=5,
)
],
lineno=6,
col_offset=5,
),
comment=Comment(s="# comment 6", lineno=6, col_offset=11),
elsecomment=None,
lineno=6,
col_offset=5,
)
],
lineno=2,
col_offset=0,
),
comment=Comment(s="# comment 2", lineno=2, col_offset=10),
elsecomment=None,
lineno=2,
col_offset=0,
),
Comment(s="# comment 18", lineno=18, col_offset=0),
]
)
check_add_comments(code, exp)
def test_gather_names_name():
node = Name(id='y', ctx=Store())
exp = {'y'}
obs = ast.gather_names(node)
assert_equal(exp, obs)
