From eab87afb1c9323b83da3710f34a05d14feb56b2e Mon Sep 17 00:00:00 2001 From: Gaurav Sheni Date: Fri, 14 Oct 2022 13:04:29 -0400 Subject: [PATCH] TST: Fix unsigned pyarrow types in SIGNED_INT_PYARROW_DTYPES (#48457) * Update __init__.py * Update __init__.py * Update __init__.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py * Update test_arrow.py Co-authored-by: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> --- pandas/_testing/__init__.py | 2 +- pandas/tests/extension/test_arrow.py | 7 +++++++ 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/pandas/_testing/__init__.py b/pandas/_testing/__init__.py index adc52a506fcaf0..ce6b6ceea56040 100644 --- a/pandas/_testing/__init__.py +++ b/pandas/_testing/__init__.py @@ -199,7 +199,7 @@ import pyarrow as pa UNSIGNED_INT_PYARROW_DTYPES = [pa.uint8(), pa.uint16(), pa.uint32(), pa.uint64()] - SIGNED_INT_PYARROW_DTYPES = [pa.uint8(), pa.int16(), pa.int32(), pa.uint64()] + SIGNED_INT_PYARROW_DTYPES = [pa.int8(), pa.int16(), pa.int32(), pa.int64()] ALL_INT_PYARROW_DTYPES = UNSIGNED_INT_PYARROW_DTYPES + SIGNED_INT_PYARROW_DTYPES FLOAT_PYARROW_DTYPES = [pa.float32(), pa.float64()] diff --git a/pandas/tests/extension/test_arrow.py b/pandas/tests/extension/test_arrow.py index 92e4dbaea4eea8..c6915e491554a2 100644 --- a/pandas/tests/extension/test_arrow.py +++ b/pandas/tests/extension/test_arrow.py @@ -1624,6 +1624,13 @@ def test_add_series_with_extension_array(self, data, request): reason=f"add_checked not implemented for {pa_dtype}", ) ) + elif pa_dtype.equals("int8"): + request.node.add_marker( + pytest.mark.xfail( + raises=pa.ArrowInvalid, + reason=f"raises on overflow for {pa_dtype}", + ) + ) super().test_add_series_with_extension_array(data)