TST: add test case from Issue pandas-dev#10989

Author: jakevdp

pandas/tools/pivot.py

@@ -170,7 +170,8 @@ def convert_categorical(ind):
             return _convert(ind)
 
     table.index = convert_categorical(table.index)
-    table.columns = convert_categorical(table.columns)
+    if hasattr(table, 'columns'):
+        table.columns = convert_categorical(table.columns)
 
     if not values and isinstance(table, Series):
         # If there are no values and the table is a series, then there is only

pandas/tools/tests/test_pivot.py

@@ -719,6 +719,20 @@ def test_crosstab_dropna(self):
                                     ('two', 'dull'), ('two', 'shiny')])
         assert_equal(res.columns.values, m.values)
 
+    def test_categorical_margins(self):
+        # GH 10989
+        data = pd.DataFrame({'x': np.arange(8),
+                             'y': np.arange(8) // 4,
+                             'z': np.arange(8) % 2})
+        data.y = data.y.astype('category')
+        data.z = data.z.astype('category')
+        table = data.pivot_table('x', 'y', 'z', margins=True)
+        assert_equal(table.values, [[1, 2, 1.5],
+                                    [5, 6, 5.5],
+                                    [3, 4, 3.5]])
+                                    
+        
+
 if __name__ == '__main__':
     import nose
     nose.runmodule(argv=[__file__, '-vvs', '-x', '--pdb', '--pdb-failure'],