Skip to content

Commit a60fbb0

Browse files
committed
Updated test case to include other operators which included fill_value
1 parent 5ac26a4 commit a60fbb0

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

pandas/tests/frame/test_arithmetic.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2182,7 +2182,8 @@ def test_mixed_col_index_dtype(using_infer_string):
21822182
tm.assert_frame_equal(result, expected)
21832183

21842184

2185-
def test_df_mul_series_fill_value():
2185+
@pytest.mark.parametrize("op", ["add", "sub", "mul", "div", "mod", "truediv", "pow"])
2186+
def test_df_series_fill_value(op):
21862187
# GH 61581
21872188
data = np.arange(50).reshape(10, 5)
21882189
columns = list("ABCDE")
@@ -2196,7 +2197,9 @@ def test_df_mul_series_fill_value():
21962197
df_b = df.iloc[:, -1]
21972198
nan_mask = df_a.isna().astype(int).mul(df_b.isna().astype(int), axis=0).astype(bool)
21982199

2199-
df_result = df_a.mul(df_b, axis=0, fill_value=5)
2200-
df_expected = (df_a.fillna(5).mul(df_b.fillna(5), axis=0)).mask(nan_mask, np.nan)
2200+
df_result = getattr(df_a, op)(df_b, axis=0, fill_value=5)
2201+
df_expected = getattr(df_a.fillna(5), op)(df_b.fillna(5), axis=0).mask(
2202+
nan_mask, np.nan
2203+
)
22012204

22022205
tm.assert_frame_equal(df_result, df_expected)

0 commit comments

Comments
 (0)