1 files changed, 6 insertions, 11 deletions
diff --git a/test/test_unpack_raw.py b/test/test_unpack_raw.py
index 15d9c93..9f3784c 100644
--- a/test/test_unpack_raw.py
+++ b/test/test_unpack_raw.py
@@ -1,7 +1,5 @@
 """Tests for cases where the user seeks to obtain packed msgpack objects"""
 
-from nose import main
-from nose.tools import *
 import six
 from msgpack import Unpacker, packb
 
@@ -10,14 +8,14 @@ def test_write_bytes():
     unpacker = Unpacker()
     unpacker.feed(b'abc')
     f = six.BytesIO()
-    assert_equal(unpacker.unpack(f.write), ord('a'))
-    assert_equal(f.getvalue(), b'a')
+    assert unpacker.unpack(f.write) == ord('a')
+    assert f.getvalue() == b'a'
     f = six.BytesIO()
     assert unpacker.skip(f.write) is None
-    assert_equal(f.getvalue(), b'b')
+    assert f.getvalue() == b'b'
     f = six.BytesIO()
     assert unpacker.skip() is None
-    assert_equal(f.getvalue(), b'')
+    assert f.getvalue() == b''
 
 
 def test_write_bytes_multi_buffer():
@@ -27,8 +25,5 @@ def test_write_bytes_multi_buffer():
 
     f = six.BytesIO()
     unpacked = unpacker.unpack(f.write)
-    assert_equal(unpacked, long_val)
-    assert_equal(f.getvalue(), expected)
-
-if __name__ == '__main__':
-    main()
+    assert unpacked == long_val
+    assert f.getvalue() == expected