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

Add files

Browse source 
Signed-off-by: András Schmelczer <andras@schmelczer.dev>
This commit is contained in:
Andras Schmelczer 2022-04-02 13:47:33 +02:00
commit 889e79174b
No known key found for this signature in database
GPG key ID: 39260B5B0614A13E
103 changed files with 24322 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

42
good_ai/tests/sus/test_parallel_map.py Normal file
View file

@ -0,0 +1,42 @@
import unittest
from src.sus.parallel_map import parallel_map
COUNT = int(1e5) + 3
class TestParallelMap(unittest.TestCase):
def test_simple_case_with_progress_bar(self) -> None:
inputs = range(COUNT)
expected = [v**2 for v in range(COUNT)]
self.assertEqual(parallel_map(lambda v: v**2, inputs), expected)
def test_simple_case_without_progress_bar(self) -> None:
inputs = range(COUNT)
expected = [v**2 for v in range(COUNT)]
self.assertEqual(
parallel_map(lambda v: v**2, inputs, disable_progress=True), expected
)
def test_simple_case_invalid_values(self) -> None:
inputs = range(COUNT)
self.assertRaises(
AssertionError, parallel_map, lambda v: v**2, inputs, concurrency=0
)
self.assertRaises(
AssertionError, parallel_map, lambda v: v**2, inputs, chunk_size=0
)
def test_no_op(self) -> None:
self.assertEqual(parallel_map(lambda v: v**2, [], disable_progress=True), [])
self.assertEqual(
parallel_map(lambda v: v**2, [], disable_progress=True, chunk_size=100),
[],
)
self.assertEqual(
parallel_map(lambda v: v**2, [], disable_progress=True, concurrency=100),
[],
)