@@ -164,7 +164,7 @@ def nested(): # pylint: disable=unused-variable
164
164
165
165
166
166
class TestAssignedVarAnalysis (unittest .TestCase ):
167
- def assertDefs (self , f , expected : set [str ]):
167
+ def assert_assigned_vars (self , f , expected : set [str ]):
168
168
source , parse_tree = ast_utils .get_src_and_ast (f )
169
169
result = analysis .assigned_vars (parse_tree .body , formatter (source ))
170
170
self .assertEqual (result , expected )
@@ -175,7 +175,7 @@ def f(x):
175
175
y = x + 2
176
176
return y
177
177
178
- self .assertDefs (f , {"x" , "y" })
178
+ self .assert_assigned_vars (f , {"x" , "y" })
179
179
180
180
def test_if_defs (self ):
181
181
def f (x ):
@@ -187,7 +187,7 @@ def f(x):
187
187
z = 3 * t
188
188
return z
189
189
190
- self .assertDefs (f , {"z" , "y" , "t" })
190
+ self .assert_assigned_vars (f , {"z" , "y" , "t" })
191
191
192
192
def test_loop_defs (self ):
193
193
def f (x ):
@@ -198,7 +198,7 @@ def f(x):
198
198
sum = sum + square
199
199
return sum
200
200
201
- self .assertDefs (f , {"sum" , "x" , "square" })
201
+ self .assert_assigned_vars (f , {"sum" , "x" , "square" })
202
202
203
203
def test_if_loop_defs (self ):
204
204
def f (x ):
@@ -212,7 +212,7 @@ def f(x):
212
212
sum = 0
213
213
return sum
214
214
215
- self .assertDefs (f , {"sum" , "x" , "square" })
215
+ self .assert_assigned_vars (f , {"sum" , "x" , "square" })
216
216
217
217
218
218
if __name__ == "__main__" :
0 commit comments