diff --git a/python/pyarrow/tests/test_array.py b/python/pyarrow/tests/test_array.py index 009ab1e849b..9a5044ce394 100644 --- a/python/pyarrow/tests/test_array.py +++ b/python/pyarrow/tests/test_array.py @@ -550,7 +550,8 @@ def test_arange(): for case in cases: result = pa.arange(*case) result.validate(full=True) - assert result.equals(pa.array(list(range(*case)), type=pa.int64())) + + assert result.equals(pa.array(list(range(*case)), type=pa.int64())) # type: ignore[no-matching-overload] # Validate memory_pool keyword argument result = pa.arange(-1, 101, memory_pool=pa.default_memory_pool()) diff --git a/python/pyarrow/tests/test_compute.py b/python/pyarrow/tests/test_compute.py index ad61dbc48a7..97f694df1fc 100644 --- a/python/pyarrow/tests/test_compute.py +++ b/python/pyarrow/tests/test_compute.py @@ -1737,10 +1737,10 @@ def test_arithmetic_multiply(): @pytest.mark.parametrize("ty", ["round", "round_to_multiple"]) def test_round_to_integer(ty): if ty == "round": - round = pc.round + round_func = pc.round RoundOptions = partial(pc.RoundOptions, ndigits=0) elif ty == "round_to_multiple": - round = pc.round_to_multiple + round_func = pc.round_to_multiple RoundOptions = partial(pc.RoundToMultipleOptions, multiple=1) values = [3.2, 3.5, 3.7, 4.5, -3.2, -3.5, -3.7, None] @@ -1758,7 +1758,7 @@ def test_round_to_integer(ty): } for round_mode, expected in rmode_and_expected.items(): options = RoundOptions(round_mode=round_mode) - result = round(values, options=options) + result = round_func(values, options=options) expected_array = pa.array(expected, type=pa.float64()) assert expected_array.equals(result)