TST: fix tests xref pandas-dev#13554 (pandas-dev#16054)

analyticalmonk · Apr 19, 2017 · 00063d2 · 00063d2
1 parent 0b22b8d
commit 00063d2
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pandas/tests/test_categorical.py b/pandas/tests/test_categorical.py
@@ -3043,9 +3043,10 @@ def test_pivot_table(self):
             [Categorical(["a", "b", "z"], ordered=True),
              Categorical(["c", "d", "y"], ordered=True)],
             names=['A', 'B'])
-        expected = Series([1, 2, np.nan, 3, 4, np.nan, np.nan, np.nan, np.nan],
-                          index=exp_index, name='values')
-        tm.assert_series_equal(result, expected)
+        expected = DataFrame(
+            {'values': [1, 2, np.nan, 3, 4, np.nan, np.nan, np.nan, np.nan]},
+            index=exp_index)
+        tm.assert_frame_equal(result, expected)
 
     def test_count(self):