--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/tests/test_prettysize.py	Sat Sep 29 14:01:47 2012 +0200
@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+
+from pgtoolkit 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 input, result in expected:
+            self.assertEqual(psize(input), result)
+
+
+if __name__ == '__main__':
+    unittest.main()
+