1 files changed, 10 insertions, 2 deletions
diff --git a/test/test_utils.py b/test/test_utils.py
index 0cdd3fae..942136fd 100644
--- a/test/test_utils.py
+++ b/test/test_utils.py
@@ -5,9 +5,10 @@ import tutils
 def test_bidi():
     b = utils.BiDi(a=1, b=2)
     assert b.a == 1
-    assert b[1] == "a"
+    assert b.get_name(1) == "a"
+    assert b.get_name(5) is None
     tutils.raises(AttributeError, getattr, b, "c")
-    tutils.raises(KeyError, b.__getitem__, 5)
+    tutils.raises(ValueError, utils.BiDi, one=1, two=1)
 
 
 def test_hexdump():
@@ -19,3 +20,10 @@ def test_cleanBin():
     assert utils.cleanBin("\00ne") == ".ne"
     assert utils.cleanBin("\nne") == "\nne"
     assert utils.cleanBin("\nne", True) == ".ne"
+
+
+def test_pretty_size():
+    assert utils.pretty_size(100) == "100B"
+    assert utils.pretty_size(1024) == "1kB"
+    assert utils.pretty_size(1024 + (1024/2.0)) == "1.5kB"
+    assert utils.pretty_size(1024*1024) == "1MB"