Skip to content

Commit 5c66bd1

Browse files
Fix test
1 parent 3e93acc commit 5c66bd1

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

Lib/test/test_urllibnet.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -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

230232
if __name__ == "__main__":

0 commit comments

Comments
 (0)