File tree Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Expand file tree Collapse file tree 1 file changed +4
-6
lines changed Original file line number Diff line number Diff 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'\n Index: { i } \n Length: { dataset_length } \n Threads: { thread_count } \n Expected: { (i , b )} \n Actual: { 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'\n Length: { dataset_length } \n Threads: { thread_count } \n Expected: { heap } \n Actual: { algorithm .chunk_split (dataset_length , thread_count )} '
You can’t perform that action at this time.
0 commit comments