Backport PR #48457 on branch 1.5.x (TST: Fix unsigned pyarrow types in SIGNED_INT_PYARROW_DTYPES) (#49093)

meeseeksmachine · gsheni · web-flow · commit c58f2057b8c8 · 2022-10-14T21:27:30.000+02:00
Backport PR #48457: TST: Fix unsigned pyarrow types in SIGNED_INT_PYARROW_DTYPES Co-authored-by: Gaurav Sheni <gvsheni@gmail.com>
diff --git a/pandas/_testing/__init__.py b/pandas/_testing/__init__.py
@@ -197,7 +197,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
@@ -1570,6 +1570,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)
 
 

Original file line number	Diff line number	Diff line change
`@@ -1570,6 +1570,13 @@ def test_add_series_with_extension_array(self, data, request):`
`1570`	`1570`	`reason=f"add_checked not implemented for {pa_dtype}",`
`1571`	`1571`	`)`
`1572`	`1572`	`)`
	`1573`	`+ elif pa_dtype.equals("int8"):`
	`1574`	`+ request.node.add_marker(`
	`1575`	`+ pytest.mark.xfail(`
	`1576`	`+ raises=pa.ArrowInvalid,`
	`1577`	`+ reason=f"raises on overflow for {pa_dtype}",`
	`1578`	`+ )`
	`1579`	`+ )`
`1573`	`1580`	`super().test_add_series_with_extension_array(data)`
`1574`	`1581`
`1575`	`1582`