@@ -221,12 +221,14 @@ def recording_reporthook(blocks, block_size, total_size):
221221 self .assertEqual (records [0 ][2 ], expected_size )
222222 self .assertEqual (records [- 1 ][2 ], expected_size )
223223
224- block_sizes = {block_size for _ , block_size , _ in records }
225- self .assertEqual ({records [0 ][1 ]}, block_sizes ,
226- msg = "block sizes in %s must be equal" % records_repr )
227- self .assertGreaterEqual (records [- 1 ][0 ]* records [0 ][1 ], expected_size ,
228- msg = "number of blocks * block size must be"
229- " >= total size in %s" % records_repr )
224+ self .assertEqual (records [0 ][1 ], 8192 ,
225+ msg = "first block size should be 8192 in %s" % records_repr )
226+ for block_num , block_size , total_size in records :
227+ self .assertLessEqual (block_size , 8192 ,
228+ msg = "block size should be <= 8192 in %s" % records_repr )
229+ total_read = sum (block_size for _ , block_size , _ in records [1 :])
230+ self .assertEqual (total_read , expected_size ,
231+ msg = "sum of bytes read must equal total size in %s" % records_repr )
230232
231233
232234if __name__ == "__main__" :
0 commit comments