-
Notifications
You must be signed in to change notification settings - Fork 10
Expand file tree
/
Copy pathtest_clean_code.py
More file actions
94 lines (70 loc) · 1.73 KB
/
test_clean_code.py
File metadata and controls
94 lines (70 loc) · 1.73 KB
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
from libcst.codemod import CodemodTest
from codemodder.utils.clean_code import RemoveEmptyExpressionsFormatting
class TestRemoveEmptyExpressionsFormatting(CodemodTest):
TRANSFORM = RemoveEmptyExpressionsFormatting
def test_empty_string(self):
before = """
"string" % ""
"""
after = """
"string"
"""
self.assertCodemod(before, after)
def test_empty_dict(self):
before = """
"string" % {}
"""
after = """
"string"
"""
self.assertCodemod(before, after)
def test_empty_tuple(self):
before = """
"string" % ()
"""
after = """
"string"
"""
self.assertCodemod(before, after)
def test_single_dict_removal(self):
before = """
other_d = {'c':""}
d = {'a':1, 'b':2, **other_d}
"%(a)s%(b)s%(c)s" % d
"""
after = """
other_d = {}
d = {'a':1, 'b':2, **other_d}
"%(a)s%(b)s" % d
"""
self.assertCodemod(before, after)
def test_single_tuple_removal(self):
before = """
t = (1, "", 3,)
"%s%s%s" % t
"""
after = """
t = (1, 3,)
"%s%s" % t
"""
self.assertCodemod(before, after)
def test_remove_all_tuple(self):
before = """
t = ("", "", "",)
"%s%s%s" % t
"""
after = """
t = ()
''
"""
self.assertCodemod(before, after)
def test_remove_all_dict(self):
before = """
d = {'a':"", 'b':""}
"%(a)s%(b)s" % d
"""
after = """
d = {}
''
"""
self.assertCodemod(before, after)