Fixed test for python3.

eerimoq · Aug 29, 2016 · a8ca4e0 · a8ca4e0
1 parent 6069a11
commit a8ca4e0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/bitstruct.py b/bitstruct.py
@@ -1,7 +1,7 @@
 import re
 import struct
 
-__version__ = "3.3.0"
+__version__ = "3.3.1"
 
 
 def _parse_format(fmt):

diff --git a/tests/test_bitstruct.py b/tests/test_bitstruct.py
@@ -1,5 +1,6 @@
 import unittest
 import timeit
+import sys
 from bitstruct import *
 
 
@@ -67,9 +68,14 @@ def test_pack(self):
             pack('f32', "foo")
             self.fail()
         except ValueError as e:
-            self.assertEqual(
-                str(e),
-                'could not convert string to float: foo')
+            if sys.version_info[0] < 3:
+                self.assertEqual(
+                    str(e),
+                    'could not convert string to float: foo')
+            else:
+                self.assertEqual(
+                    str(e),
+                    "could not convert string to float: 'foo'")
 
         # Cannot convert argument to bytearray.
         try: