Skip to content

Commit 32a0f77

Browse files
committed
Updated test case to include other operators which included fill_value
1 parent 5421761 commit 32a0f77

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
@@ -2175,7 +2175,8 @@ def test_mixed_col_index_dtype(string_dtype_no_object):
21752175
tm.assert_frame_equal(result, expected)
21762176

21772177

2178-
def test_df_mul_series_fill_value():
2178+
@pytest.mark.parametrize("op", ["add", "sub", "mul", "div", "mod", "truediv", "pow"])
2179+
def test_df_series_fill_value(op):
21792180
# GH 61581
21802181
data = np.arange(50).reshape(10, 5)
21812182
columns = list("ABCDE")
@@ -2189,7 +2190,9 @@ def test_df_mul_series_fill_value():
21892190
df_b = df.iloc[:, -1]
21902191
nan_mask = df_a.isna().astype(int).mul(df_b.isna().astype(int), axis=0).astype(bool)
21912192

2192-
df_result = df_a.mul(df_b, axis=0, fill_value=5)
2193-
df_expected = (df_a.fillna(5).mul(df_b.fillna(5), axis=0)).mask(nan_mask, np.nan)
2193+
df_result = getattr(df_a, op)(df_b, axis=0, fill_value=5)
2194+
df_expected = getattr(df_a.fillna(5), op)(df_b.fillna(5), axis=0).mask(
2195+
nan_mask, np.nan
2196+
)
21942197

21952198
tm.assert_frame_equal(df_result, df_expected)

0 commit comments

Comments
 (0)