Skip to content

Commit 45ffc77

Browse files
committed
Fixes pandas-dev#24893: re-added test
1 parent d0a04f8 commit 45ffc77

File tree

1 file changed

+20
-1
lines changed

1 file changed

+20
-1
lines changed

pandas/tests/reshape/test_pivot.py

+20-1
Original file line numberDiff line numberDiff line change
@@ -1630,6 +1630,25 @@ def test_categorical_margins_category(self, observed):
16301630
table = df.pivot_table("x", "y", "z", dropna=observed, margins=True)
16311631
tm.assert_frame_equal(table, expected)
16321632

1633+
def test_margins_casted_to_float(self, observed):
1634+
# GH #24893
1635+
df = pd.DataFrame(
1636+
{
1637+
"A": [2, 4, 6, 8],
1638+
"B": [1, 4, 5, 8],
1639+
"C": [1, 3, 4, 6],
1640+
"D": ["X", "X", "Y", "Y"],
1641+
}
1642+
)
1643+
1644+
result = pd.pivot_table(df, index="D", margins=True)
1645+
expected = pd.DataFrame(
1646+
{"A": [3, 7, 5], "B": [2.5, 6.5, 4.5], "C": [2, 5, 3.5]},
1647+
index=pd.Index(["X", "Y", "All"], name="D"),
1648+
)
1649+
table = result
1650+
tm.assert_frame_equal(result, expected)
1651+
16331652
def test_categorical_aggfunc(self, observed):
16341653
# GH 9534
16351654
df = pd.DataFrame(
@@ -2527,4 +2546,4 @@ def test_margin_normalize(self):
25272546
codes=[[1, 1, 2, 2, 0], [1, 2, 1, 2, 0]],
25282547
names=["A", "B"],
25292548
)
2530-
tm.assert_frame_equal(result, expected)
2549+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)