Skip to content

Commit 1993c7c

Browse files
author
Daniel Saxton
committed
Parametrize tests
1 parent 633a6b8 commit 1993c7c

File tree

1 file changed

+4
-19
lines changed

1 file changed

+4
-19
lines changed

pandas/tests/frame/test_apply.py

+4-19
Original file line numberDiff line numberDiff line change
@@ -116,28 +116,13 @@ def test_apply_with_reduce_empty(self):
116116
# Ensure that x.append hasn't been called
117117
assert x == []
118118

119-
def test_apply_funcs_over_empty(self):
119+
@pytest.mark.parametrize("func", ["sum", "prod", "any", "all"])
120+
def test_apply_funcs_over_empty(self, func):
120121
# GH 28213
121122
df = DataFrame(columns=["a", "b", "c"])
122123

123-
result = df.apply(np.sum)
124-
expected = df.sum()
125-
assert_series_equal(result, expected)
126-
127-
result = df.apply(np.prod)
128-
expected = df.prod()
129-
assert_series_equal(result, expected)
130-
131-
result = df.apply(np.any)
132-
expected = df.any()
133-
assert_series_equal(result, expected)
134-
135-
result = df.apply(np.all)
136-
expected = df.all()
137-
assert_series_equal(result, expected)
138-
139-
result = df.nunique()
140-
expected = Series(0, index=df.columns)
124+
result = df.apply(eval(func))
125+
expected = operator.methodcaller(func)(df)
141126
assert_series_equal(result, expected)
142127

143128
def test_apply_deprecate_reduce(self):

0 commit comments

Comments
 (0)