unify tests for py2 and py3

1 files changed, 15 insertions, 13 deletions

diff --git a/test/test_sequnpack.py b/test/test_sequnpack.py
index d61be23..5fd377c 100644
--- a/test/test_sequnpack.py
+++ b/test/test_sequnpack.py
@@ -1,33 +1,35 @@
 #!/usr/bin/env python
 # coding: utf-8
 
+
+
 from msgpack import Unpacker
 
 def test_foobar():
     unpacker = Unpacker(read_size=3)
-    unpacker.feed('foobar')
-    assert unpacker.unpack() == ord('f')
-    assert unpacker.unpack() == ord('o')
-    assert unpacker.unpack() == ord('o')
-    assert unpacker.unpack() == ord('b')
-    assert unpacker.unpack() == ord('a')
-    assert unpacker.unpack() == ord('r')
+    unpacker.feed(b'foobar')
+    assert unpacker.unpack() == ord(b'f')
+    assert unpacker.unpack() == ord(b'o')
+    assert unpacker.unpack() == ord(b'o')
+    assert unpacker.unpack() == ord(b'b')
+    assert unpacker.unpack() == ord(b'a')
+    assert unpacker.unpack() == ord(b'r')
     try:
         o = unpacker.unpack()
-        print "Oops!", o
+        print(("Oops!", o))
         assert 0
     except StopIteration:
         assert 1
     else:
         assert 0
-    unpacker.feed('foo')
-    unpacker.feed('bar')
+    unpacker.feed(b'foo')
+    unpacker.feed(b'bar')
 
     k = 0
-    for o, e in zip(unpacker, 'foobarbaz'):
-        assert o == ord(e)
+    for o, e in zip(unpacker, b'foobarbaz'):
+        assert o == e
         k += 1
-    assert k == len('foobar')
+    assert k == len(b'foobar')
 
 if __name__ == '__main__':
     test_foobar()