Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 2eb4c19

Browse files
narendasannarendasan
authoredApr 30, 2024
chore(//tests): Update tests to use assertEqual (#2800)
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent dc96072 commit 2eb4c19

File tree

6 files changed

+53
-53
lines changed

6 files changed

+53
-53
lines changed
 

‎tests/py/dynamo/backend/test_backend_compiler.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ def forward(self, x, y):
2222
fx_graph = torch.fx.symbolic_trace(FullySupportedMultiOp())
2323
partitioned_graph, _ = fast_partition(deepcopy(fx_graph), min_block_size=3)
2424

25-
self.assertEquals(
25+
self.assertEqual(
2626
len(
2727
[
2828
1
@@ -96,17 +96,17 @@ def forward(self, x, y):
9696
testing_partitioning=True,
9797
)
9898

99-
self.assertEquals(
99+
self.assertEqual(
100100
len(unexpected_ops_seen),
101101
0,
102102
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
103103
)
104-
self.assertEquals(
104+
self.assertEqual(
105105
len(partitioned_graphs),
106106
1,
107107
"Without control flow breaks, there should only be a single graph",
108108
)
109-
self.assertEquals(
109+
self.assertEqual(
110110
len(
111111
[
112112
1
@@ -201,7 +201,7 @@ def forward(self, x, y):
201201
fx_graph = torch.fx.symbolic_trace(FullySupportedMultiOp())
202202
partitioned_graph, _ = fast_partition(deepcopy(fx_graph), min_block_size=3)
203203

204-
self.assertEquals(
204+
self.assertEqual(
205205
len(list(partitioned_graph.named_children())),
206206
1,
207207
"All operators are supported, there should be one segment",
@@ -265,17 +265,17 @@ def forward(self, x, y):
265265
testing_partitioning=True,
266266
)
267267

268-
self.assertEquals(
268+
self.assertEqual(
269269
len(unexpected_ops_seen),
270270
0,
271271
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
272272
)
273-
self.assertEquals(
273+
self.assertEqual(
274274
len(partitioned_graphs),
275275
1,
276276
"Without control flow breaks, there should only be a single graph",
277277
)
278-
self.assertEquals(
278+
self.assertEqual(
279279
len(
280280
[
281281
1

‎tests/py/dynamo/backend/test_specialized_models.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def forward(self, x):
3333
min_block_size=1,
3434
)
3535

36-
self.assertEquals(
36+
self.assertEqual(
3737
len(expected_ops_unseen),
3838
0,
3939
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -90,7 +90,7 @@ def forward(self, x):
9090
min_block_size=1,
9191
)
9292

93-
self.assertEquals(
93+
self.assertEqual(
9494
len(expected_ops_unseen),
9595
0,
9696
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -263,13 +263,13 @@ def forward(self, x):
263263
min_block_size=1,
264264
)
265265

266-
self.assertEquals(
266+
self.assertEqual(
267267
len(unexpected_ops_seen),
268268
0,
269269
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
270270
)
271271

272-
self.assertEquals(
272+
self.assertEqual(
273273
len(expected_ops_unseen),
274274
0,
275275
f"The following expected ops were not encountered: {expected_ops_unseen}",

‎tests/py/dynamo/lowering/test_aten_lowering_passes.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -118,13 +118,13 @@ def forward(self, x):
118118
min_block_size=1,
119119
)
120120

121-
self.assertEquals(
121+
self.assertEqual(
122122
len(unexpected_ops_seen),
123123
0,
124124
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
125125
)
126126

127-
self.assertEquals(
127+
self.assertEqual(
128128
len(expected_ops_unseen),
129129
0,
130130
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -187,13 +187,13 @@ def forward(self, q, k, v):
187187
min_block_size=1,
188188
)
189189

190-
self.assertEquals(
190+
self.assertEqual(
191191
len(unexpected_ops_seen),
192192
0,
193193
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
194194
)
195195

196-
self.assertEquals(
196+
self.assertEqual(
197197
len(expected_ops_unseen),
198198
0,
199199
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -303,13 +303,13 @@ def forward(self, q, k, v):
303303
min_block_size=1,
304304
)
305305

306-
self.assertEquals(
306+
self.assertEqual(
307307
len(unexpected_ops_seen),
308308
0,
309309
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
310310
)
311311

312-
self.assertEquals(
312+
self.assertEqual(
313313
len(expected_ops_unseen),
314314
0,
315315
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -418,13 +418,13 @@ def forward(self, input, weight, bias):
418418
min_block_size=1,
419419
)
420420

421-
self.assertEquals(
421+
self.assertEqual(
422422
len(unexpected_ops_seen),
423423
0,
424424
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
425425
)
426426

427-
self.assertEquals(
427+
self.assertEqual(
428428
len(expected_ops_unseen),
429429
0,
430430
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -524,13 +524,13 @@ def forward(self, input):
524524
min_block_size=1,
525525
)
526526

527-
self.assertEquals(
527+
self.assertEqual(
528528
len(unexpected_ops_seen),
529529
0,
530530
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
531531
)
532532

533-
self.assertEquals(
533+
self.assertEqual(
534534
len(expected_ops_unseen),
535535
0,
536536
f"The following expected ops were not encountered: {expected_ops_unseen}",

‎tests/py/dynamo/lowering/test_decompositions.py

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def forward(self, x, y):
3434
fx_graph, inputs, expected_ops=expected_ops, min_block_size=2
3535
)
3636

37-
self.assertEquals(
37+
self.assertEqual(
3838
len(expected_ops_unseen),
3939
0,
4040
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -63,7 +63,7 @@ def forward(self, x):
6363
fx_graph, inputs, unexpected_ops=unexpected_ops, min_block_size=1
6464
)
6565

66-
self.assertEquals(
66+
self.assertEqual(
6767
len(unexpected_ops_seen),
6868
0,
6969
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
@@ -98,13 +98,13 @@ def forward(self, x):
9898
min_block_size=1,
9999
)
100100

101-
self.assertEquals(
101+
self.assertEqual(
102102
len(unexpected_ops_seen),
103103
0,
104104
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
105105
)
106106

107-
self.assertEquals(
107+
self.assertEqual(
108108
len(expected_ops_unseen),
109109
0,
110110
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -138,13 +138,13 @@ def forward(self, x):
138138
min_block_size=1,
139139
)
140140

141-
self.assertEquals(
141+
self.assertEqual(
142142
len(unexpected_ops_seen),
143143
0,
144144
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
145145
)
146146

147-
self.assertEquals(
147+
self.assertEqual(
148148
len(expected_ops_unseen),
149149
0,
150150
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -206,13 +206,13 @@ def forward(self, x):
206206
min_block_size=1,
207207
)
208208

209-
self.assertEquals(
209+
self.assertEqual(
210210
len(unexpected_ops_seen),
211211
0,
212212
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
213213
)
214214

215-
self.assertEquals(
215+
self.assertEqual(
216216
len(expected_ops_unseen),
217217
0,
218218
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -265,13 +265,13 @@ def forward(self, x):
265265
min_block_size=1,
266266
)
267267

268-
self.assertEquals(
268+
self.assertEqual(
269269
len(unexpected_ops_seen),
270270
0,
271271
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
272272
)
273273

274-
self.assertEquals(
274+
self.assertEqual(
275275
len(expected_ops_unseen),
276276
0,
277277
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -325,13 +325,13 @@ def forward(self, x):
325325
min_block_size=1,
326326
)
327327

328-
self.assertEquals(
328+
self.assertEqual(
329329
len(unexpected_ops_seen),
330330
0,
331331
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
332332
)
333333

334-
self.assertEquals(
334+
self.assertEqual(
335335
len(expected_ops_unseen),
336336
0,
337337
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -385,13 +385,13 @@ def forward(self, x):
385385
min_block_size=1,
386386
)
387387

388-
self.assertEquals(
388+
self.assertEqual(
389389
len(unexpected_ops_seen),
390390
0,
391391
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
392392
)
393393

394-
self.assertEquals(
394+
self.assertEqual(
395395
len(expected_ops_unseen),
396396
0,
397397
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -449,13 +449,13 @@ def forward(self, x):
449449
min_block_size=1,
450450
)
451451

452-
self.assertEquals(
452+
self.assertEqual(
453453
len(unexpected_ops_seen),
454454
0,
455455
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
456456
)
457457

458-
self.assertEquals(
458+
self.assertEqual(
459459
len(expected_ops_unseen),
460460
0,
461461
f"The following expected ops were not encountered: {expected_ops_unseen}",
@@ -478,7 +478,7 @@ def forward(self, x):
478478
optimized_model_results_shape = optimized_model_results.size()
479479
torch_model_results_shape = torch_model_results.size()
480480

481-
self.assertEquals(
481+
self.assertEqual(
482482
optimized_model_results_shape,
483483
torch_model_results_shape,
484484
f"The optimized model results shape and torch model results shape should be equal in empty_like",

‎tests/py/dynamo/partitioning/test_fast_partitioning.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def forward(self, x, y):
1919

2020
fx_graph = torch.fx.symbolic_trace(FullySupportedOneOp())
2121
partitioned_graph, _ = partitioning.fast_partition(deepcopy(fx_graph))
22-
self.assertEquals(
22+
self.assertEqual(
2323
len(
2424
[
2525
1
@@ -43,7 +43,7 @@ def forward(self, x, y):
4343
partitioned_graph, _ = partitioning.fast_partition(
4444
deepcopy(fx_graph), require_full_compilation=True
4545
)
46-
self.assertEquals(
46+
self.assertEqual(
4747
len(
4848
[
4949
1
@@ -71,7 +71,7 @@ def forward(self, x, y):
7171
partitioned_graph, _ = partitioning.fast_partition(
7272
deepcopy(fx_graph), min_block_size=2
7373
)
74-
self.assertEquals(
74+
self.assertEqual(
7575
len(
7676
[
7777
1
@@ -100,7 +100,7 @@ def forward(self, x, y):
100100
partitioned_graph, _ = partitioning.fast_partition(
101101
deepcopy(fx_graph), min_block_size=2
102102
)
103-
self.assertEquals(
103+
self.assertEqual(
104104
len(
105105
[
106106
1
@@ -155,18 +155,18 @@ def forward(self, x, y):
155155
use_fast_partitioner=True,
156156
)
157157

158-
self.assertEquals(
158+
self.assertEqual(
159159
len(unexpected_ops_seen),
160160
0,
161161
f"The following unexpected ops were encountered: {unexpected_ops_seen}",
162162
)
163163

164-
self.assertEquals(
164+
self.assertEqual(
165165
len(partitioned_graphs),
166166
1,
167167
"Without control flow breaks, there should only be a single graph",
168168
)
169-
self.assertEquals(
169+
self.assertEqual(
170170
len(
171171
[
172172
1

0 commit comments

Comments
 (0)