diff -r d3ffa15f5886 -r cc27136cdead tests/test_prettysize.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tests/test_prettysize.py Sat Apr 06 00:07:17 2013 +0200 @@ -0,0 +1,29 @@ +#!/usr/bin/env python3 + +from pycolib import prettysize + +import unittest + + +class TestPrettySize(unittest.TestCase): + + def test_prettysize(self): + psize = prettysize.prettysize_short + expected = ( + (0, '0'), + (1000, '1000'), + (1024, '1K'), + (1050, '1.03K'), + (2050, '2K'), + (333333, '325.52K'), + (1048576, '1M'), + (1050000, '1M'), + (5555555, '5.3M'), + (1073741824, '1G'), + ) + for value, result in expected: + self.assertEqual(psize(value), result) + + +if __name__ == '__main__': + unittest.main()