diff -r eaae9539e910 -r efee419b7a2d tests/testprettysize.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tests/testprettysize.py Mon Jul 18 17:39:37 2011 +0200 @@ -0,0 +1,26 @@ +#!/usr/bin/env python3 + +import unittest +import prettysize + +class TestHumansize(unittest.TestCase): + def test_humansize(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 input, result in expected: + self.assertEqual(psize(input), result) + + +if __name__ == '__main__': + unittest.main()