Skip to content

Commit 67077ab

Browse files
test(perf): Check as we generate instead of writing all of chunk to memory
1 parent 20d702d commit 67077ab

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

tests/test_algorithm.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -37,15 +37,13 @@ def test_chunking_dynamic():
3737
expected_chunk_high = dataset_length % thread_count
3838

3939
i = 0
40-
heap = []
40+
gen = algorithm.chunk_split(dataset_length, thread_count)
4141
while i < dataset_length:
4242
chunk_length = expected_chunk_length_low + int(expected_chunk_high > 0)
4343
b = i + chunk_length
4444

45-
heap.append((i, b))
45+
assert (
46+
next(gen) == (i, b)
47+
), f'\nIndex: {i}\nLength: {dataset_length}\nThreads: {thread_count}\nExpected: {(i, b)}\nActual: {next(gen)}'
4648
expected_chunk_high -= 1
4749
i = b
48-
49-
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

Comments
 (0)