improve tests

pandas-dev · Feb 23, 2017 · 215c2aa · 215c2aa
1 parent cd9354f
commit 215c2aa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pandas/tests/io/test_packers.py b/pandas/tests/io/test_packers.py
@@ -123,10 +123,10 @@ 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)
+        df2 = df.astype({0: 'category'}).set_index(0)
+        s = to_msgpack(None, df2)
         result = read_msgpack(s)
-        tm.assert_frame_equal(result, df)
+        tm.assert_frame_equal(result, df2)
 
         with ensure_clean(self.path) as p: