TST: Verify operators with IntegerArray and list-likes (22606) (#35987)

Author: avinashpancham

pandas/tests/arithmetic/conftest.py

@@ -2,6 +2,7 @@
 import pytest
 
 import pandas as pd
+import pandas._testing as tm
 
 # ------------------------------------------------------------------
 # Helper Functions
@@ -230,5 +231,14 @@ def box_with_array(request):
     return request.param
 
 
+@pytest.fixture(params=[pd.Index, pd.Series, tm.to_array, np.array, list], ids=id_func)
+def box_1d_array(request):
+    """
+    Fixture to test behavior for Index, Series, tm.to_array, numpy Array and list
+    classes
+    """
+    return request.param
+
+
 # alias so we can use the same fixture for multiple parameters in a test
 box_with_array2 = box_with_array

pandas/tests/arithmetic/test_numeric.py

@@ -11,11 +11,19 @@
 import pytest
 
 import pandas as pd
-from pandas import Index, Series, Timedelta, TimedeltaIndex
+from pandas import Index, Int64Index, Series, Timedelta, TimedeltaIndex, array
 import pandas._testing as tm
 from pandas.core import ops
 
 
+@pytest.fixture(params=[Index, Series, tm.to_array])
+def box_pandas_1d_array(request):
+    """
+    Fixture to test behavior for Index, Series and tm.to_array classes
+    """
+    return request.param
+
+
 def adjust_negative_zero(zero, expected):
     """
     Helper to adjust the expected result if we are dividing by -0.0
@@ -1330,3 +1338,33 @@ def test_dataframe_div_silenced():
     )
     with tm.assert_produces_warning(None):
         pdf1.div(pdf2, fill_value=0)
+
+
+@pytest.mark.parametrize(
+    "data, expected_data",
+    [([0, 1, 2], [0, 2, 4])],
+)
+def test_integer_array_add_list_like(
+    box_pandas_1d_array, box_1d_array, data, expected_data
+):
+    # GH22606 Verify operators with IntegerArray and list-likes
+    arr = array(data, dtype="Int64")
+    container = box_pandas_1d_array(arr)
+    left = container + box_1d_array(data)
+    right = box_1d_array(data) + container
+
+    if Series == box_pandas_1d_array:
+        assert_function = tm.assert_series_equal
+        expected = Series(expected_data, dtype="Int64")
+    elif Series == box_1d_array:
+        assert_function = tm.assert_series_equal
+        expected = Series(expected_data, dtype="object")
+    elif Index in (box_pandas_1d_array, box_1d_array):
+        assert_function = tm.assert_index_equal
+        expected = Int64Index(expected_data)
+    else:
+        assert_function = tm.assert_numpy_array_equal
+        expected = np.array(expected_data, dtype="object")
+
+    assert_function(left, expected)
+    assert_function(right, expected)