def test_nasted():
assert get_insertion_indexes(
dedent("""
x = (
(1, 2),
(3, 4)
)
""")) == [29]
assert get_insertion_indexes(
dedent("""
x = (
1,
(
2,
3
),
(
3,
(
4,
5
)
)
)
""")) == [56, 123, 133, 139]
def test_dict():
assert get_insertion_indexes(
dedent("""
x = {
'a': 1,
'b': 2
}
""")) == [29]
assert get_insertion_indexes(
dedent("""
x = {
'a': 1,
'b': 2 # comment
}
""")) == [29]
assert get_insertion_indexes(
dedent("""
x = {
'a': 1, 'b': 2
}
""")) == [25]
assert get_insertion_indexes(
dedent("""
function_call({
'a': 1,
'b': 2
})
""")) == [39]
def test_nasted():
assert get_insertion_indexes(
dedent("""
x = {
{1, 2},
{3, 4}
}
""")) == [29]
assert get_insertion_indexes(
dedent("""
x = {
1,
{
2,
3
},
{
3,
{
4,
5
}
}
}
""")) == [56, 123, 133, 139]
def test_nasted():
assert get_insertion_indexes(
dedent("""
x = [
[1, 2],
[3, 4]
]
""")) == [29]
assert get_insertion_indexes(
dedent("""
x = [
1,
[
2,
3
],
[
3,
[
4,
5
]
]
]
""")) == [56, 123, 133, 139]
def test_classdef():
assert get_insertion_indexes(
dedent("""
class Test(
a,
b
): pass
""")) == [25]
assert get_insertion_indexes(
dedent("""
class Test(
a,
b # comment
): pass
""")) == [25]
assert get_insertion_indexes(
dedent("""
class Test(
a,
metaclass=b
): pass
""")) == [35]
assert get_insertion_indexes(
dedent("""
class Test(
a,
not_metaclass=b
): pass
""")) == [39]
def test_no_add_ignore():
assert get_insertion_indexes(dedent("""
async def test(): pass
""")) == []
assert get_insertion_indexes(dedent("""
async def test(
): pass
""")) == []
assert get_insertion_indexes(dedent("""
async def test(
): pass
""")) == []
def test_no_add_exists():
assert get_insertion_indexes(
dedent("""
function_call(
a,
b,
c,
d,
)
""")) == []
assert get_insertion_indexes(
dedent("""
function_call(
a
,
)
""")) == []
assert get_insertion_indexes(
dedent("""
function_call(
a,
b,
c,
d, # comment
)
""")) == []
assert get_insertion_indexes(
dedent("""
function_call(
one(
),
two(
three(),
),
four(
five(
seven(
eight,
),
),
),
)
""")) == []
def test_tuple():
assert get_insertion_indexes(
dedent("""
x = (
1,
2
)
""")) == [19]
assert get_insertion_indexes(
dedent("""
x = (
1,
2 # comment
)
""")) == [19]
assert get_insertion_indexes(
dedent("""
x = (
1, 2
)
""")) == [15]
assert get_insertion_indexes(
dedent("""
(
a,
b
) = (
1,
2
)
""")) == [15, 34]
assert get_insertion_indexes(
dedent("""
a, b = (
1,
2
)
""")) == [22]
assert get_insertion_indexes(
dedent("""
function_call((
1,
2,
3
))
""")) == [36]
assert get_insertion_indexes(
dedent("""
Type[
1,
2
]
""")) == [19]
def test_decorator_call():
assert get_insertion_indexes(
dedent("""
@decorator(
a,
b
)
def function(a, b):
pass
""")) == [25]
def test_list():
assert get_insertion_indexes(
dedent("""
x = [
1,
2
]
""")) == [19]
assert get_insertion_indexes(
dedent("""
x = [
1,
2 # comment
]
""")) == [19]
assert get_insertion_indexes(
dedent("""
x = [
1, 2
]
""")) == [15]
assert get_insertion_indexes(
dedent("""
[
a,
b
] = [
1,
2
]
""")) == [15, 34]
assert get_insertion_indexes(
dedent("""
function_call([
1,
2,
3
])
""")) == [36]
def test_no_add_exists():
assert get_insertion_indexes(
dedent("""
[
1,
]
""")) == []
assert get_insertion_indexes(
dedent("""
[
1
,
]
""")) == []
assert get_insertion_indexes(
dedent("""
x = [
1,
2,
]
""")) == []
assert get_insertion_indexes(
dedent("""
x = [
1,
2, # comment
]
""")) == []
assert get_insertion_indexes(
dedent("""
x = [
1, 2,
]
""")) == []
assert get_insertion_indexes(
dedent("""
[
a,
b,
] = [
1,
2,
]
""")) == []
assert get_insertion_indexes(
dedent("""
function_call([
1,
2,
3,
])
""")) == []
def test_tuple():
assert get_insertion_indexes(
dedent("""
x = {
1,
2
}
""")) == [19]
assert get_insertion_indexes(
dedent("""
x = {
1,
2 # comment
}
""")) == [19]
assert get_insertion_indexes(
dedent("""
x = {
1, 2
}
""")) == [15]
assert get_insertion_indexes(
dedent("""
a, b = {
1,
2
}
""")) == [22]
assert get_insertion_indexes(
dedent("""
function_call({
1,
2,
3
})
""")) == [36]
def test_no_add_ignore():
assert get_insertion_indexes(
dedent("""
class Test(object): pass
""")) == []
assert get_insertion_indexes(dedent("""
class Test(): pass
""")) == []
assert get_insertion_indexes(
dedent("""
class Test(
): pass
""")) == []
assert get_insertion_indexes(
dedent("""
class Test(
): pass
""")) == []
assert get_insertion_indexes(dedent("""
class Test: pass
""")) == []
assert get_insertion_indexes(
dedent("""
class Test(a, b, c): pass
""")) == []
def test_no_add_exists():
assert get_insertion_indexes(
dedent("""
class Test(
a,
b,
): pass
""")) == []
assert get_insertion_indexes(
dedent("""
class Test(
a
,
): pass
""")) == []
assert get_insertion_indexes(
dedent("""
class Test(
a,
b, # comment
): pass
""")) == []
assert get_insertion_indexes(
dedent("""
class Test(
a,
metaclass=b,
): pass
""")) == []
assert get_insertion_indexes(
dedent("""
class Test(
a,
not_metaclass=b,
): pass
""")) == []
def test_no_add_ignore():
assert get_insertion_indexes(
dedent("""
function_call(a, b,
c, d)
""")) == []
assert get_insertion_indexes(
dedent("""
function_call(a, b, c, d)
""")) == []
assert get_insertion_indexes(dedent("""
function_call()
""")) == []
assert get_insertion_indexes(
dedent("""
function_call(
)
""")) == []
assert get_insertion_indexes(
dedent("""
function_call(
)
""")) == []
assert get_insertion_indexes(
dedent("""
function_call(
# A comment
)
""")) == []
def test_no_add_generator_as_single_param():
assert get_insertion_indexes(
dedent("""
function_call(
100 for _ in range(10)
)
""")) == []
assert get_insertion_indexes(
dedent("""
function_call(
1 + 1
for _ in range(10)
)
""")) == []
assert get_insertion_indexes(
dedent("""
function_call(
1 + 1
for _ in range(10)
if True
)
""")) == []
def test_no_add_ignore():
assert get_insertion_indexes(dedent("""
x = set()
""")) == []
assert get_insertion_indexes(dedent("""
x = {
}
""")) == []
assert get_insertion_indexes(dedent("""
x = {
}
""")) == []
assert get_insertion_indexes(
dedent("""
x = {
# Comment
}
""")) == []
assert get_insertion_indexes(dedent("""
x = {1, 2}
""")) == []
assert get_insertion_indexes(
dedent("""
x = {1, 2,
3, 4}
""")) == []
assert get_insertion_indexes(dedent("""
(a, b) = {1, 2}
""")) == []
assert get_insertion_indexes(
dedent("""
function_call({1, 2, 3})
""")) == []
def test_no_add_ignore():
assert get_insertion_indexes(dedent("""
x = []
""")) == []
assert get_insertion_indexes(dedent("""
x = [
]
""")) == []
assert get_insertion_indexes(dedent("""
x = [
]
""")) == []
assert get_insertion_indexes(
dedent("""
x = [
# Comment
]
""")) == []
assert get_insertion_indexes(dedent("""
x = [1, 2]
""")) == []
assert get_insertion_indexes(
dedent("""
x = [1, 2,
3, 4]
""")) == []
assert get_insertion_indexes(dedent("""
[a, b] = [1, 2]
""")) == []
assert get_insertion_indexes(
dedent("""
function_call([1, 2, 3])
""")) == []
def test_no_add_exists():
assert get_insertion_indexes(
dedent("""
x = (
1,
2,
)
""")) == []
assert get_insertion_indexes(
dedent("""
x = (
1, 2,
)
""")) == []
assert get_insertion_indexes(
dedent("""
x = (
1
,
)
""")) == []
assert get_insertion_indexes(
dedent("""
(
a,
b,
) = (
1,
2,
)
""")) == []
assert get_insertion_indexes(
dedent("""
function_call((
1,
2,
3,
))
""")) == []
assert get_insertion_indexes(
dedent("""
Type[
1,
2,
]
""")) == []
def test_nasted():
assert get_insertion_indexes(
dedent("""
x = {
'a': {
'1': 1,
'2': 2
},
'b': {
'3': 3
},
'c': {
'd': {
'e': {
'f': 'g'
}
}
}
}
""")) == [48, 81, 158, 172, 182, 188]
def test_no_add_exists():
assert get_insertion_indexes(
dedent("""
{
1,
}
""")) == []
assert get_insertion_indexes(
dedent("""
{
1
,
}
""")) == []
assert get_insertion_indexes(
dedent("""
x = {
1,
2,
}
""")) == []
assert get_insertion_indexes(
dedent("""
x = {
1, 2,
}
""")) == []
assert get_insertion_indexes(
dedent("""
a, b = {
1,
2,
}
""")) == []
assert get_insertion_indexes(
dedent("""
function_call({
1,
2,
3,
})
""")) == []
def test_no_add_exists():
assert get_insertion_indexes(
dedent("""
{
1: 1,
}
""")) == []
assert get_insertion_indexes(
dedent("""
{
1: 1
,
}
""")) == []
assert get_insertion_indexes(
dedent("""
x = {
'a': 1,
'b': 2,
}
""")) == []
assert get_insertion_indexes(
dedent("""
x = {
'a': 1, 'b': 2,
}
""")) == []
assert get_insertion_indexes(
dedent("""
function_call({
'a': 1,
'b': 2,
})
""")) == []
assert get_insertion_indexes(
dedent("""
function_call({
'a': 1,
'b': 2, # comment
})
""")) == []
def test_no_add_ignore():
assert get_insertion_indexes(dedent("""
x = {}
""")) == []
assert get_insertion_indexes(dedent("""
x = {
}
""")) == []
assert get_insertion_indexes(dedent("""
x = {
}
""")) == []
assert get_insertion_indexes(
dedent("""
x = {
# Comment
}
""")) == []
assert get_insertion_indexes(
dedent("""
x = {'1': 2, '3': 4}
""")) == []
assert get_insertion_indexes(
dedent("""
x = {'1': 2,
'3': 4}
""")) == []
assert get_insertion_indexes(
dedent("""
function_call({'1': 2, '3': 4})
""")) == []
def test_no_add_exists():
assert get_insertion_indexes(dedent("""
async def test(
a,
): pass
""")) == []
assert get_insertion_indexes(dedent("""
async def test(
param=value,
): pass
""")) == []
assert get_insertion_indexes(dedent("""
async def test(
*args,
): pass
""")) == []
assert get_insertion_indexes(dedent("""
async def test(
a,
param=value,
): pass
""")) == []
assert get_insertion_indexes(dedent("""
async def test(
a,
*,
param=value,
): pass
""")) == []
assert get_insertion_indexes(dedent("""
async def test(
*,
param=value,
): pass
""")) == []
assert get_insertion_indexes(dedent("""
async def test(
a,
*args,
param=value,
): pass
""")) == []
assert get_insertion_indexes(dedent("""
async def test(
a,
*args,
param=value,
**kwargs,
): pass
""")) == []
assert get_insertion_indexes(dedent("""
async def test(
a,
*args,
param=value,
**kwargs, # comment
): pass
""")) == []
def test_functiondef():
assert get_insertion_indexes(
dedent("""
def test(
a
): pass
""")) == [16]
assert get_insertion_indexes(
dedent("""
def test(
param=value
): pass
""")) == [26]
assert get_insertion_indexes(
dedent("""
def test(
*args
): pass
""")) == [20]
assert get_insertion_indexes(
dedent("""
def test(
a,
param=value
): pass
""")) == [33]
assert get_insertion_indexes(
dedent("""
def test(
a,
*,
param=value
): pass
""")) == [40]
assert get_insertion_indexes(
dedent("""
def test(
*,
param=value
): pass
""")) == [33]
assert get_insertion_indexes(
dedent("""
def test(
a,
*args,
param=value
): pass
""")) == [44]
assert get_insertion_indexes(
dedent("""
def test(
a,
*args,
param=value,
**kwargs
): pass
""")) == [58]
assert get_insertion_indexes(
dedent("""
def test(
a,
*args,
param=value,
**kwargs # comment
): pass
""")) == [58]
assert get_insertion_indexes(
dedent("""
def test(
a=((),(),())
): pass
""")) == [27]
assert get_insertion_indexes(
dedent("""
def test(
a=b() + c(1) + d()
): pass
""")) == [33]
assert get_insertion_indexes(
dedent("""
def test(
a=b[c:d]
): pass
""")) == [23]
def test_no_add_ignore():
assert get_insertion_indexes(dedent("""
x = []
""")) == []
assert get_insertion_indexes(dedent("""
x = (
)
""")) == []
assert get_insertion_indexes(dedent("""
x = (
)
""")) == []
assert get_insertion_indexes(
dedent("""
x = (
# Comment
)
""")) == []
assert get_insertion_indexes(dedent("""
x = (1, 2)
""")) == []
assert get_insertion_indexes(
dedent("""
x = (1, 2,
3, 4)
""")) == []
assert get_insertion_indexes(dedent("""
a, b = 1, 2
""")) == []
assert get_insertion_indexes(dedent("""
a, = 1, 2
""")) == []
assert get_insertion_indexes(dedent("""
(a, b) = (1, 2)
""")) == []
assert get_insertion_indexes(
dedent("""
function_call((1, 2, 3))
""")) == []
assert get_insertion_indexes(dedent("""
Type[1]
""")) == []
assert get_insertion_indexes(dedent("""
Type[1, 2]
""")) == []
assert get_insertion_indexes(
dedent("""
Type[1, 2,
3, 4]
""")) == []
def test_call():
assert get_insertion_indexes(
dedent("""
function_call(
a,
b
)
""")) == [28]
assert get_insertion_indexes(
dedent("""
function_call(
a,
b # comment
)
""")) == [28]
assert get_insertion_indexes(
dedent("""
function_call(
a,
b,
c=d
)
""")) == [37]
assert get_insertion_indexes(
dedent("""
function_call(
*args,
**kwargs
)
""")) == [39]
assert get_insertion_indexes(
dedent("""
function_call(a,
b
)
""")) == [33]
assert get_insertion_indexes(
dedent("""
function_call(a,
b,
c=d
)
""")) == [52]
assert get_insertion_indexes(
dedent("""
function_call(another_call(
a,
b
))
""")) == [41]
assert get_insertion_indexes(
dedent("""
function_call(
another_call(
a,
b=c
),
dd=ff
)
""")) == [56, 73]
assert get_insertion_indexes(
dedent("""
obj.method(
a,
b
)
""")) == [25]
assert get_insertion_indexes(
dedent("""
function_call(
(1 for _ in range(10)),
(2 for _ in range(10))
)
""")) == [70]
assert get_insertion_indexes(
dedent("""
function_call(
one(
),
two(
three()
),
four(
five(
seven(
eight=8
)
)
)
)
""")) == [56, 130, 144, 154, 160]
def test_functiondef():
assert get_insertion_indexes(dedent("""
async def test(
a
): pass
""")) == [22]
assert get_insertion_indexes(dedent("""
async def test(
param=value
): pass
""")) == [32]
assert get_insertion_indexes(dedent("""
async def test(
*args
): pass
""")) == [26]
assert get_insertion_indexes(dedent("""
async def test(
a,
param=value
): pass
""")) == [39]
assert get_insertion_indexes(dedent("""
async def test(
a,
*,
param=value
): pass
""")) == [46]
assert get_insertion_indexes(dedent("""
async def test(
*,
param=value
): pass
""")) == [39]
assert get_insertion_indexes(dedent("""
async def test(
a,
*args,
param=value
): pass
""")) == [50]
assert get_insertion_indexes(dedent("""
async def test(
a,
*args,
param=value,
**kwargs
): pass
""")) == [64]
assert get_insertion_indexes(dedent("""
async def test(
a,
*args,
param=value,
**kwargs # comment
): pass
""")) == [64]
assert get_insertion_indexes(dedent("""
async def test(
a=((),(),())
): pass
""")) == [33]
assert get_insertion_indexes(dedent("""
async def test(
a=b() + c(1) + d()
): pass
""")) == [39]
assert get_insertion_indexes(dedent("""
async def test(
a=b[c:d]
): pass
""")) == [29]
