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