improve tests

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)