frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE] Enable ruff's UP rules and autoformat test/ (#105434)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/105434
Approved by: https://github.com/albanD
This commit is contained in:
Justin Chu
2023-07-19 07:40:18 -07:00
committed by PyTorch MergeBot
parent 7b56238551
commit 73e1455327
88 changed files with 427 additions and 451 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_sort_and_select.py
View File

@ -33,12 +33,12 @@ class TestSortAndSelect(TestCase):
# see above
return ((b != b) | (a <= b)).all().item()
else:
error('unknown order "{}", must be "ascending" or "descending"'.format(order))
error(f'unknown order "{order}", must be "ascending" or "descending"')
are_ordered = True
for k in range(1, SIZE):
self.assertTrue(check_order(mxx[:, k - 1], mxx[:, k]),
'torch.sort ({}) values unordered for {}'.format(order, task))
f'torch.sort ({order}) values unordered for {task}')
seen = set()
indicesCorrect = True
@ -51,7 +51,7 @@ class TestSortAndSelect(TestCase):
seen.clear()
for j in range(size):
self.assertEqual(x[k][ixx[k][j]], mxx[k][j],
msg='torch.sort ({}) indices wrong for {}'.format(order, task))
msg=f'torch.sort ({order}) indices wrong for {task}')
seen.add(ixx[k][j])
self.assertEqual(len(seen), size)