Comments and testcase.

eerimoq · Mar 28, 2016 · 8e7056b · 8e7056b
1 parent 39b1a5d
commit 8e7056b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/bitstruct.py b/bitstruct.py
@@ -156,7 +156,7 @@ def pack(fmt, *args):
 
 
 def unpack(fmt, data):
-    """Unpack the byte string (presumably packed by pack(fmt, ...))
+    """Unpack `data` (byte string, bytearray or list of integers)
     according to the given format. The result is a tuple even if it
     contains exactly one item.
 

diff --git a/tests/test_bitstruct.py b/tests/test_bitstruct.py
@@ -45,7 +45,7 @@ def test_unpack(self):
         unpacked = unpack('u1u1s6u7u9', b'\x3e\x82\x16')
         self.assertEqual(unpacked, (0, 0, -2, 65, 22))
 
-        unpacked = unpack('u1', b'\x80')
+        unpacked = unpack('u1', bytearray(b'\x80'))
         self.assertEqual(unpacked, (1,))
 
         packed = b'\x00\x80\x00\x00\x00\x00\x08\x00\x00\x00'
@@ -64,7 +64,7 @@ def test_unpack(self):
         unpacked = unpack('u1s6f32r43', packed)
         self.assertEqual(unpacked, (0, -2, 3.75, b'\x00\xff\x00\xff\x00\xe0'))
 
-        packed = b'\x80'
+        packed = [0x80]
         unpacked = unpack('b1', packed)
         self.assertEqual(unpacked, (True,))