Skip to content

Commit 062031e

Browse files
author
Kei
committed
Remove unneccessary tests
1 parent 5a30216 commit 062031e

File tree

1 file changed

+0
-107
lines changed

1 file changed

+0
-107
lines changed

pandas/tests/groupby/test_apply.py

Lines changed: 0 additions & 107 deletions
Original file line numberDiff line numberDiff line change
@@ -1583,110 +1583,3 @@ def f_4(grp):
15831583
e.loc["Pony"] = np.nan
15841584
e.name = None
15851585
tm.assert_series_equal(result, e)
1586-
1587-
1588-
# GH58291
1589-
def test_apply_frame_not_as_index_returns_single_index():
1590-
df = DataFrame(
1591-
[
1592-
["group_a", 0],
1593-
["group_a", 2],
1594-
["group_b", 1],
1595-
["group_b", 3],
1596-
["group_b", 5],
1597-
],
1598-
columns=["group", "value"],
1599-
)
1600-
gb = df.groupby("group", as_index=False)[["group", "value"]]
1601-
1602-
def up_to_two_rows(df: DataFrame) -> DataFrame:
1603-
return df.head(2)
1604-
1605-
result = gb.apply(up_to_two_rows)
1606-
1607-
expected = DataFrame(
1608-
[["group_a", 0], ["group_a", 2], ["group_b", 1], ["group_b", 3]],
1609-
columns=["group", "value"],
1610-
index=range(4),
1611-
)
1612-
tm.assert_frame_equal(result, expected)
1613-
1614-
1615-
# GH58291
1616-
def test_apply_column_groupby_frame_not_as_index_returns_single_index(df):
1617-
gb = df.groupby("A", as_index=False)[["A", "C", "D"]]
1618-
result = gb.apply(lambda x: x.min())
1619-
print(str(result))
1620-
1621-
expected = DataFrame(
1622-
[["bar", -2.441467, -2.441467], ["foo", -0.413064, -0.413064]],
1623-
columns=["A", "C", "D"],
1624-
index=range(2),
1625-
)
1626-
tm.assert_frame_equal(result, expected)
1627-
1628-
1629-
# GH58291
1630-
def test_apply_non_column_groupby_frame_not_as_index_returns_single_index(tsframe):
1631-
gb = tsframe.groupby(lambda x: x.month, as_index=False)
1632-
result = gb.apply(lambda x: x.mean())
1633-
print(str(result))
1634-
1635-
expected = DataFrame(
1636-
[
1637-
[1, 0.115464, 0.260960, 0.187824, -0.411523],
1638-
[2, 0.047104, -0.183591, 0.330640, -0.207427],
1639-
],
1640-
columns=["index", "A", "B", "C", "D"],
1641-
index=range(2),
1642-
)
1643-
tm.assert_frame_equal(result, expected)
1644-
1645-
result = gb.apply(lambda x: x.mean() + x.max())
1646-
print(str(result))
1647-
1648-
expected = DataFrame(
1649-
[
1650-
[1, 2.490351, 2.062381, 2.244527, 0.912824],
1651-
[2, 1.918738, 2.273746, 1.759184, 0.700975],
1652-
],
1653-
columns=["index", "A", "B", "C", "D"],
1654-
index=range(2),
1655-
)
1656-
tm.assert_frame_equal(result, expected)
1657-
1658-
result = gb.apply(lambda x: 1)
1659-
print(str(result))
1660-
1661-
expected = DataFrame([[1, 1], [2, 1]], columns=["index", None], index=range(2))
1662-
tm.assert_frame_equal(result, expected)
1663-
1664-
result = gb.apply(lambda x: x.quantile([0.2, 0.38]))
1665-
print(str(result))
1666-
1667-
expected = DataFrame(
1668-
[
1669-
[-0.742155, -0.607186, -0.325423, -1.254187],
1670-
[-0.212477, -0.034416, -0.117452, -0.561536],
1671-
[-0.842537, -0.920719, -0.404496, -0.729036],
1672-
[-0.411161, -0.830757, 0.321981, -0.444410],
1673-
],
1674-
columns=["A", "B", "C", "D"],
1675-
index=[0.20, 0.38] * 2,
1676-
)
1677-
tm.assert_frame_equal(result, expected)
1678-
1679-
result = gb.apply(DataFrame.quantile, [0.2, 0.38])
1680-
print(str(result))
1681-
1682-
expected = DataFrame(
1683-
[
1684-
[-0.742155, -0.607186, -0.325423, -1.254187],
1685-
[-0.212477, -0.034416, -0.117452, -0.561536],
1686-
[-0.842537, -0.920719, -0.404496, -0.729036],
1687-
[-0.411161, -0.830757, 0.321981, -0.444410],
1688-
],
1689-
columns=["A", "B", "C", "D"],
1690-
index=[0.20, 0.38] * 2,
1691-
)
1692-
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)