Merge pull request #4 from brunovianarezende/master

Fix error caused by missing StructError import
klbostee · Nov 13, 2012 · 2e1a806 · 2e1a806
2 parents d315a50 + 21b53da
commit 2e1a806
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/ctypedbytes.py b/ctypedbytes.py
@@ -19,6 +19,7 @@ def classes():
     import sys
     import typedbytes
     from fastb import reads, writes
+    from struct import error as StructError
 
 
     def flatten(iterable):

diff --git a/tests/testio.py b/tests/testio.py
@@ -3,7 +3,7 @@
 import decimal
 import datetime
 import ctypedbytes as typedbytes
-
+from struct import error as StructError
 
 class TestIO(unittest.TestCase):
 
@@ -25,6 +25,20 @@ def testio(self):
         file.close()
         os.remove("test.bin")
 
+    def testwrongio(self):
+        try:
+            file = open("test.bin", "wb")
+            output = typedbytes.Output(file)
+            output.writes([1])
+            file.close()
+            file = open("test.bin", "rb")
+            input = typedbytes.Input(file)
+            input = typedbytes.PairedInput(file)
+            self.assertRaises(StructError, lambda :list(input.reads()))
+            file.close()
+        finally:
+            os.remove("test.bin")
+
     def testpairio(self):
         objects = TestIO.objects
         file = open("test.bin", "wb")