Skip to content

Commit 4107a96

Browse files
committed
TST: add tests for inplace ops
1 parent 2eda0c6 commit 4107a96

File tree

1 file changed

+60
-0
lines changed

1 file changed

+60
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,60 @@
1+
import pytest
2+
3+
from pandas import Series
4+
import pandas._testing as tm
5+
6+
7+
@pytest.mark.parametrize(
8+
"ser1, ser2, expected_add, expected_sub, expected_mul",
9+
(
10+
[
11+
Series([1], dtype="Int64"),
12+
Series([2], dtype="Int64"),
13+
Series([3], dtype="Int64"),
14+
Series([-1], dtype="Int64"),
15+
Series([2], dtype="Int64"),
16+
],
17+
[
18+
Series([1], dtype="float"),
19+
Series([2.0], dtype="float"),
20+
Series([3.0], dtype="float"),
21+
Series([-1.0], dtype="float"),
22+
Series([2.0], dtype="float"),
23+
],
24+
[
25+
Series([1], dtype="Int64"),
26+
Series([2.0], dtype="float"),
27+
Series([3.0], dtype="float"),
28+
Series([-1.0], dtype="float"),
29+
Series([2], dtype="float"),
30+
],
31+
[
32+
Series([1.0], dtype="float"),
33+
Series([2], dtype="Int64"),
34+
Series([3.0], dtype="float"),
35+
Series([-1.0], dtype="float"),
36+
Series([2], dtype="float"),
37+
],
38+
pytest.param(
39+
Series([1], dtype="Int64"),
40+
Series([2.0], dtype="Float64"),
41+
Series([3.0], dtype="Float64"),
42+
Series([-1.0], dtype="Float64"),
43+
Series([2], dtype="Float64"),
44+
marks=pytest.mark.xfail(reason="Not implemented yet"),
45+
),
46+
),
47+
)
48+
def test_series_inplace_ops(ser1, ser2, expected_add, expected_sub, expected_mul):
49+
50+
res = ser1.copy()
51+
res += ser2
52+
tm.assert_series_equal(res, expected_add)
53+
54+
res = ser1.copy()
55+
res -= ser2
56+
tm.assert_series_equal(res, expected_sub)
57+
58+
res = ser1.copy()
59+
res *= ser2
60+
tm.assert_series_equal(res, expected_mul)

0 commit comments

Comments
 (0)