Skip to content

Commit 4c8da73

Browse files
committed
Rename method
1 parent 3026408 commit 4c8da73

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

onnxscript/analysis_test.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def nested(): # pylint: disable=unused-variable
164164

165165

166166
class TestAssignedVarAnalysis(unittest.TestCase):
167-
def assertDefs(self, f, expected: set[str]):
167+
def assert_assigned_vars(self, f, expected: set[str]):
168168
source, parse_tree = ast_utils.get_src_and_ast(f)
169169
result = analysis.assigned_vars(parse_tree.body, formatter(source))
170170
self.assertEqual(result, expected)
@@ -175,7 +175,7 @@ def f(x):
175175
y = x + 2
176176
return y
177177

178-
self.assertDefs(f, {"x", "y"})
178+
self.assert_assigned_vars(f, {"x", "y"})
179179

180180
def test_if_defs(self):
181181
def f(x):
@@ -187,7 +187,7 @@ def f(x):
187187
z = 3 * t
188188
return z
189189

190-
self.assertDefs(f, {"z", "y", "t"})
190+
self.assert_assigned_vars(f, {"z", "y", "t"})
191191

192192
def test_loop_defs(self):
193193
def f(x):
@@ -198,7 +198,7 @@ def f(x):
198198
sum = sum + square
199199
return sum
200200

201-
self.assertDefs(f, {"sum", "x", "square"})
201+
self.assert_assigned_vars(f, {"sum", "x", "square"})
202202

203203
def test_if_loop_defs(self):
204204
def f(x):
@@ -212,7 +212,7 @@ def f(x):
212212
sum = 0
213213
return sum
214214

215-
self.assertDefs(f, {"sum", "x", "square"})
215+
self.assert_assigned_vars(f, {"sum", "x", "square"})
216216

217217

218218
if __name__ == "__main__":

0 commit comments

Comments
 (0)