else/Bytestring/test_bytestring.py
Ethan Dalool 53645b0123 else
2016-08-09 01:33:36 -07:00

30 lines
No EOL
1 KiB
Python

import bytestring
import unittest
pairs = {
100: '100.000 b',
2 ** 10: '1.000 KiB',
2 ** 20: '1.000 MiB',
2 ** 30: '1.000 GiB',
-(2 ** 30): '-1.000 GiB',
(2 ** 30) + (512 * (2 ** 20)): '1.500 GiB',
}
class BytestringTest(unittest.TestCase):
def test_bytestring(self):
for (number, text) in pairs.items():
self.assertEqual(bytestring.bytestring(number), text)
def test_parsebytes(self):
for (number, text) in pairs.items():
self.assertEqual(bytestring.parsebytes(text), number)
self.assertEqual(bytestring.parsebytes('100k'), 102400)
self.assertEqual(bytestring.parsebytes('100 k'), 102400)
self.assertEqual(bytestring.parsebytes('100 kb'), 102400)
self.assertEqual(bytestring.parsebytes('100 kib'), 102400)
self.assertEqual(bytestring.parsebytes('100.00KB'), 102400)
self.assertEqual(bytestring.parsebytes('1.5 mb'), 1572864)
self.assertEqual(bytestring.parsebytes('-1.5 mb'), -1572864)
if __name__ == '__main__':
unittest.main()