fix test

Author: abast

pandas/tests/msgpack/test_unpack.py

@@ -3,7 +3,7 @@
 from pandas.msgpack import Unpacker, packb, OutOfData, ExtType
 import pandas.util.testing as tm
 import pytest
-from pandas import DataFrame
+from pandas import DataFrame, read_msgpack
 
 class TestUnpack(tm.TestCase):
 
@@ -65,11 +65,11 @@ def _hook(self, code, data):
 
     def test_unpack_categorical_index(self):
         '''dataframe with CategoricalIndex can be read and written'''
-        pdf = pd.DataFrame(dict(A=[1,1,1,2,2,2], B = [1,2,3,4,5,6]))
+        pdf = DataFrame(dict(A=[1,1,1,2,2,2], B = [1,2,3,4,5,6]))
         pdf['A'] = pdf['A'].astype('category')
         pdf.set_index('A', inplace = True)
         f = BytesIO()
         pdf.to_msgpack(f)
         f.seek(0)
-        pdf2 = pd.read_msgpack(f)
+        pdf2 = read_msgpack(f)
         tm.assert_frame_equal(pdf, pdf2)