SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Some tests misusing assertTrue for comparisons fix (#16771)

Browse Source 
* Fix issue avoid-misusing-assert-true found at https://codereview.doctor

* fix tests

* fix tf

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
This commit is contained in:
code-review-doctor
2022-04-19 13:44:08 +01:00
committed by GitHub
parent d3bd9ac728
commit a2392415e9
11 changed files with 30 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/trainer/test_trainer_utils.py
View File

@@ -97,9 +97,9 @@ class TrainerUtilsTest(unittest.TestCase):
gatherer.add_arrays([predictions[indices], [predictions[indices], predictions[indices]]])
result = gatherer.finalize()
self.assertTrue(isinstance(result, list))
self.assertTrue(len(result), 2)
self.assertEqual(len(result), 2)
self.assertTrue(isinstance(result[1], list))
self.assertTrue(len(result[1]), 2)
self.assertEqual(len(result[1]), 2)
self.assertTrue(np.array_equal(result[0], predictions))
self.assertTrue(np.array_equal(result[1][0], predictions))
self.assertTrue(np.array_equal(result[1][1], predictions))