Adapt test_get_handle_pyarrow_compat for pyarrow 19

Author: mroeschke

pandas/tests/io/test_common.py

@@ -27,6 +27,7 @@
 
 import pandas as pd
 import pandas._testing as tm
+from pandas.util.version import Version
 
 import pandas.io.common as icom
 
@@ -139,6 +140,7 @@ def test_bytesiowrapper_returns_correct_bytes(self):
 
     # Test that pyarrow can handle a file opened with get_handle
     def test_get_handle_pyarrow_compat(self):
+        pa = pytest.importorskip("pyarrow")
         pa_csv = pytest.importorskip("pyarrow.csv")
 
         # Test latin1, ucs-2, and ucs-4 chars
@@ -152,8 +154,8 @@ def test_get_handle_pyarrow_compat(self):
         s = StringIO(data)
         with icom.get_handle(s, "rb", is_text=False) as handles:
             df = pa_csv.read_csv(handles.handle).to_pandas()
-            # TODO will have to update this when pyarrow' to_pandas() is fixed
-            expected = expected.astype("object")
+            if Version(pa.__version__) < Version("19.0"):
+                expected = expected.astype("object")
             tm.assert_frame_equal(df, expected)
             assert not s.closed