Fixes pandas-dev#24893: updated tests

Author: mabelvj

pandas/tests/reshape/test_pivot.py

@@ -1601,21 +1601,22 @@ def test_pivot_table_margins_name_with_aggfunc_list(self):
         expected = pd.DataFrame(table.values, index=ix, columns=cols)
         tm.assert_frame_equal(table, expected)
 
-    @pytest.mark.xfail(reason="GH#17035 (np.mean of ints is casted back to ints)")
+
     def test_categorical_margins(self, observed):
         # GH 10989
         df = pd.DataFrame(
             {"x": np.arange(8), "y": np.arange(8) // 4, "z": np.arange(8) % 2}
         )
 
-        expected = pd.DataFrame([[1.0, 2.0, 1.5], [5, 6, 5.5], [3, 4, 3.5]])
+        expected = pd.DataFrame([[1, 2, 1.5], [5, 6, 5.5], [3, 4, 3.5]])
         expected.index = Index([0, 1, "All"], name="y")
         expected.columns = Index([0, 1, "All"], name="z")
 
         table = df.pivot_table("x", "y", "z", dropna=observed, margins=True)
         tm.assert_frame_equal(table, expected)
 
-    def test_margins_casted_to_float(self):
+
+    def test_margins_casted_to_float(self, observed):
         # GH #24893
         df = pd.DataFrame(
             {
@@ -1631,21 +1632,22 @@ def test_margins_casted_to_float(self):
             {"A": [3, 7, 5], "B": [2.5, 6.5, 4.5], "C": [2, 5, 3.5]},
             index=pd.Index(["X", "Y", "All"], name="D"),
         )
+        table = result
         tm.assert_frame_equal(result, expected)
 
-    @pytest.mark.xfail(reason="GH#17035 (np.mean of ints is casted back to ints)")
     def test_categorical_margins_category(self, observed):
         df = pd.DataFrame(
             {"x": np.arange(8), "y": np.arange(8) // 4, "z": np.arange(8) % 2}
         )
 
-        expected = pd.DataFrame([[1.0, 2.0, 1.5], [5, 6, 5.5], [3, 4, 3.5]])
+        expected = pd.DataFrame([[1, 2, 1.5], [5, 6, 5.5], [3, 4, 3.5]])
         expected.index = Index([0, 1, "All"], name="y")
         expected.columns = Index([0, 1, "All"], name="z")
 
         df.y = df.y.astype("category")
         df.z = df.z.astype("category")
         table = df.pivot_table("x", "y", "z", dropna=observed, margins=True)
+
         tm.assert_frame_equal(table, expected)
 
     def test_categorical_aggfunc(self, observed):