use torch.testing.assert_equal in TestCase.assertEqual (#67796)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/67796

Supersedes #58981.

cc mruberry

Test Plan: Imported from OSS

Reviewed By: ngimel

Differential Revision: D33542994

Pulled By: mruberry

fbshipit-source-id: 527099f5fdc154fd95ee48cd19f0a85eeec43443
(cherry picked from commit 1a58915e2cfde5c48ad77198a917872a03fd1b72)
This commit is contained in:
Philip Meier
2022-01-26 23:35:24 -08:00
committed by PyTorch MergeBot
parent de58a27769
commit d4d0ab71b3
9 changed files with 381 additions and 673 deletions

View File

@ -1744,7 +1744,7 @@ class TestNamedTensor(TestCase):
for device in get_all_device_types():
self._test_name_inference(
Tensor.expand, device=device,
args=(create('D:1'), [3]), expected_names=('D'))
args=(create('D:1'), [3]), expected_names=('D',))
self._test_name_inference(
Tensor.expand, device=device,