We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 20d702d commit 67077abCopy full SHA for 67077ab
1 file changed
tests/test_algorithm.py
@@ -37,15 +37,13 @@ def test_chunking_dynamic():
37
expected_chunk_high = dataset_length % thread_count
38
39
i = 0
40
- heap = []
+ gen = algorithm.chunk_split(dataset_length, thread_count)
41
while i < dataset_length:
42
chunk_length = expected_chunk_length_low + int(expected_chunk_high > 0)
43
b = i + chunk_length
44
45
- heap.append((i, b))
+ assert (
46
+ next(gen) == (i, b)
47
+ ), f'\nIndex: {i}\nLength: {dataset_length}\nThreads: {thread_count}\nExpected: {(i, b)}\nActual: {next(gen)}'
48
expected_chunk_high -= 1
49
i = b
-
- assert (
50
- list(algorithm.chunk_split(dataset_length, thread_count)) == heap
51
- ), f'\nLength: {dataset_length}\nThreads: {thread_count}\nExpected: {heap}\nActual: {algorithm.chunk_split(dataset_length, thread_count)}'
0 commit comments