andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Reformat asserts

Browse source
This commit is contained in:
Andras Schmelczer 2022-06-25 10:39:31 +02:00
parent 9b8b288ba6
commit cbd9f8472b
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
10 changed files with 77 additions and 77 deletions
Download patch file Download diff file Expand all files Collapse all files

4
great_ai/tests/utilities/test_parallel_map.py
View file

@ -10,7 +10,7 @@ class TestParallelMap(unittest.TestCase):
inputs = range(COUNT)
expected = [v**2 for v in range(COUNT)]
self.assertEqual(parallel_map(lambda v: v**2, inputs), expected)
assert parallel_map(lambda v: v**2, inputs) == expected
def test_simple_case_without_progress_bar(self) -> None:
inputs = range(COUNT)
@ -31,7 +31,7 @@ class TestParallelMap(unittest.TestCase):
)
def test_no_op(self) -> None:
self.assertEqual(parallel_map(lambda v: v**2, [], disable_progress=True), [])
assert parallel_map(lambda v: v**2, [], disable_progress=True) == []
self.assertEqual(
parallel_map(lambda v: v**2, [], disable_progress=True, chunk_size=100),
[],