[utils] Fix format_bytes output for Bytes (#2132)

Authored by: pukkandan, mdawar
This commit is contained in:
Pierre Mdawar 2021-12-28 03:38:31 +05:30 committed by pukkandan
parent ceb98323f2
commit f02d24d8d2
No known key found for this signature in database
GPG key ID: 0F00D95A001F4698
2 changed files with 15 additions and 2 deletions

View file

@ -37,6 +37,7 @@ from yt_dlp.utils import (
ExtractorError,
find_xpath_attr,
fix_xml_ampersands,
format_bytes,
float_or_none,
get_element_by_class,
get_element_by_attribute,
@ -1688,6 +1689,18 @@ Line 1
ll = reversed(ll)
test(ll, -15, 14, range(15))
def test_format_bytes(self):
self.assertEqual(format_bytes(0), '0.00B')
self.assertEqual(format_bytes(1000), '1000.00B')
self.assertEqual(format_bytes(1024), '1.00KiB')
self.assertEqual(format_bytes(1024**2), '1.00MiB')
self.assertEqual(format_bytes(1024**3), '1.00GiB')
self.assertEqual(format_bytes(1024**4), '1.00TiB')
self.assertEqual(format_bytes(1024**5), '1.00PiB')
self.assertEqual(format_bytes(1024**6), '1.00EiB')
self.assertEqual(format_bytes(1024**7), '1.00ZiB')
self.assertEqual(format_bytes(1024**8), '1.00YiB')
if __name__ == '__main__':
unittest.main()