Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
106 changes: 38 additions & 68 deletions pandas/tests/generic/test_finalize.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,7 @@
({"A": ["a", "b", "c"], "B": [1, 3, 5], "C": [2, 4, 6]},),
operator.methodcaller("melt", id_vars=["A"], value_vars=["B"]),
),
pytest.param((pd.DataFrame, frame_data, operator.methodcaller("map", lambda x: x))),
(pd.DataFrame, frame_data, operator.methodcaller("map", lambda x: x)),
pytest.param(
(
pd.DataFrame,
Expand All @@ -162,63 +162,47 @@
),
marks=not_implemented_mark,
),
pytest.param(
(pd.DataFrame, frame_data, operator.methodcaller("round", 2)),
),
(pd.DataFrame, frame_data, operator.methodcaller("round", 2)),
(pd.DataFrame, frame_data, operator.methodcaller("corr")),
pytest.param(
(pd.DataFrame, frame_data, operator.methodcaller("cov")),
marks=[
pytest.mark.filterwarnings("ignore::RuntimeWarning"),
],
),
pytest.param(
(
pd.DataFrame,
frame_data,
operator.methodcaller("corrwith", pd.DataFrame(*frame_data)),
),
),
pytest.param(
(pd.DataFrame, frame_data, operator.methodcaller("count")),
),
pytest.param(
(pd.DataFrame, frame_data, operator.methodcaller("nunique")),
(
pd.DataFrame,
frame_data,
operator.methodcaller("corrwith", pd.DataFrame(*frame_data)),
),
(pd.DataFrame, frame_data, operator.methodcaller("count")),
(pd.DataFrame, frame_data, operator.methodcaller("nunique")),
(pd.DataFrame, frame_data, operator.methodcaller("idxmin")),
(pd.DataFrame, frame_data, operator.methodcaller("idxmax")),
(pd.DataFrame, frame_data, operator.methodcaller("mode")),
pytest.param(
(pd.Series, [0], operator.methodcaller("mode")),
marks=not_implemented_mark,
),
pytest.param(
(
pd.DataFrame,
frame_data,
operator.methodcaller("quantile", numeric_only=True),
),
(
pd.DataFrame,
frame_data,
operator.methodcaller("quantile", numeric_only=True),
),
pytest.param(
(
pd.DataFrame,
frame_data,
operator.methodcaller("quantile", q=[0.25, 0.75], numeric_only=True),
),
(
pd.DataFrame,
frame_data,
operator.methodcaller("quantile", q=[0.25, 0.75], numeric_only=True),
),
pytest.param(
(
pd.DataFrame,
({"A": [pd.Timedelta(days=1), pd.Timedelta(days=2)]},),
operator.methodcaller("quantile", numeric_only=False),
),
(
pd.DataFrame,
({"A": [pd.Timedelta(days=1), pd.Timedelta(days=2)]},),
operator.methodcaller("quantile", numeric_only=False),
),
pytest.param(
(
pd.DataFrame,
({"A": [np.datetime64("2022-01-01"), np.datetime64("2022-01-02")]},),
operator.methodcaller("quantile", numeric_only=True),
),
(
pd.DataFrame,
({"A": [np.datetime64("2022-01-01"), np.datetime64("2022-01-02")]},),
operator.methodcaller("quantile", numeric_only=True),
),
(
pd.DataFrame,
Expand All @@ -230,18 +214,12 @@
({"A": [1]}, [pd.Timestamp("2000")]),
operator.methodcaller("to_period", freq="D"),
),
pytest.param(
(pd.DataFrame, frame_mi_data, operator.methodcaller("isin", [1])),
),
pytest.param(
(pd.DataFrame, frame_mi_data, operator.methodcaller("isin", pd.Series([1]))),
),
pytest.param(
(
pd.DataFrame,
frame_mi_data,
operator.methodcaller("isin", pd.DataFrame({"A": [1]})),
),
(pd.DataFrame, frame_mi_data, operator.methodcaller("isin", [1])),
(pd.DataFrame, frame_mi_data, operator.methodcaller("isin", pd.Series([1]))),
(
pd.DataFrame,
frame_mi_data,
operator.methodcaller("isin", pd.DataFrame({"A": [1]})),
),
(pd.DataFrame, frame_mi_data, operator.methodcaller("droplevel", "A")),
(pd.DataFrame, frame_data, operator.methodcaller("pop", "A")),
Expand All @@ -261,7 +239,7 @@
(pd.Series, [1], operator.inv),
(pd.DataFrame, frame_data, abs),
(pd.Series, [1], abs),
pytest.param((pd.DataFrame, frame_data, round)),
(pd.DataFrame, frame_data, round),
(pd.Series, [1], round),
(pd.DataFrame, frame_data, operator.methodcaller("take", [0, 0])),
(pd.DataFrame, frame_mi_data, operator.methodcaller("xs", "a")),
Expand Down Expand Up @@ -383,17 +361,13 @@
(pd.Series, ([1, 2],), operator.methodcaller("pct_change")),
(pd.DataFrame, frame_data, operator.methodcaller("pct_change")),
(pd.Series, ([1],), operator.methodcaller("transform", lambda x: x - x.min())),
pytest.param(
(
pd.DataFrame,
frame_mi_data,
operator.methodcaller("transform", lambda x: x - x.min()),
),
(
pd.DataFrame,
frame_mi_data,
operator.methodcaller("transform", lambda x: x - x.min()),
),
(pd.Series, ([1],), operator.methodcaller("apply", lambda x: x)),
pytest.param(
(pd.DataFrame, frame_mi_data, operator.methodcaller("apply", lambda x: x)),
),
(pd.DataFrame, frame_mi_data, operator.methodcaller("apply", lambda x: x)),
# Cumulative reductions
(pd.Series, ([1],), operator.methodcaller("cumsum")),
(pd.DataFrame, frame_data, operator.methodcaller("cumsum")),
Expand All @@ -402,12 +376,8 @@
(pd.DataFrame, frame_data, operator.methodcaller("any")),
marks=not_implemented_mark,
),
pytest.param(
(pd.DataFrame, frame_data, operator.methodcaller("sum")),
),
pytest.param(
(pd.DataFrame, frame_data, operator.methodcaller("std")),
),
(pd.DataFrame, frame_data, operator.methodcaller("sum")),
(pd.DataFrame, frame_data, operator.methodcaller("std")),
pytest.param(
(pd.DataFrame, frame_data, operator.methodcaller("mean")),
marks=not_implemented_mark,
Expand Down