improve tests

Author: abast

pandas/tests/io/test_packers.py

@@ -123,6 +123,11 @@ def test_string_io(self):
         result = read_msgpack(s)
         tm.assert_frame_equal(result, df)
 
+        df2 = df.astype({0:'category'}).set_index(0)
+        s = to_msgpack(None, df)
+        result = read_msgpack(s)
+        tm.assert_frame_equal(result, df)
+
         with ensure_clean(self.path) as p:
 
             s = df.to_msgpack()

pandas/tests/msgpack/test_unpack.py

@@ -3,7 +3,6 @@
 from pandas.msgpack import Unpacker, packb, OutOfData, ExtType
 import pandas.util.testing as tm
 import pytest
-from pandas import DataFrame, read_msgpack
 
 
 class TestUnpack(tm.TestCase):
@@ -63,14 +62,3 @@ def _hook(self, code, data):
         assert unpacker.unpack() == {'a': 123}
         unpacker.feed(packb({'a': ExtType(2, b'321')}, encoding='utf-8'))
         assert unpacker.unpack() == {'a': ExtType(2, b'321')}
-
-    def test_unpack_categorical_index(self):
-        '''dataframe with CategoricalIndex can be read and written'''
-        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 = read_msgpack(f)
-        tm.assert_frame_equal(pdf, pdf2)