move test to feather

Author: jorisvandenbossche

pandas/tests/io/test_feather.py

@@ -6,6 +6,8 @@
 import numpy as np
 import pytest
 
+from pandas.compat.pyarrow import pa_version_under18p0
+
 import pandas as pd
 import pandas._testing as tm
 
@@ -249,6 +251,23 @@ def test_string_inference(self, tmp_path):
         )
         tm.assert_frame_equal(result, expected)
 
+    @pytest.mark.skipif(pa_version_under18p0, reason="not supported before 18.0")
+    def test_string_inference_string_view_type(self, tmp_path):
+        # GH#54798
+        import pyarrow as pa
+        from pyarrow import feather
+
+        path = tmp_path / "string_view.parquet"
+        table = pa.table({"a": pa.array([None, "b", "c"], pa.string_view())})
+        feather.write_feather(table, path)
+
+        with pd.option_context("future.infer_string", True):
+            result = read_feather(path)
+        expected = pd.DataFrame(
+            data={"a": [None, "b", "c"]}, dtype=pd.StringDtype(na_value=np.nan)
+        )
+        tm.assert_frame_equal(result, expected)
+
     def test_out_of_bounds_datetime_to_feather(self):
         # GH#47832
         df = pd.DataFrame(

pandas/tests/io/test_parquet.py

@@ -17,7 +17,6 @@
     pa_version_under13p0,
     pa_version_under15p0,
     pa_version_under17p0,
-    pa_version_under18p0,
 )
 
 import pandas as pd
@@ -1145,26 +1144,6 @@ def test_infer_string_large_string_type(self, tmp_path, pa):
         )
         tm.assert_frame_equal(result, expected)
 
-    @pytest.mark.skipif(pa_version_under18p0, reason="not supported before 18.0")
-    def test_infer_string_string_view_type(self, tmp_path, pa):
-        # GH#54798
-        import pyarrow as pa
-        import pyarrow.parquet as pq
-
-        path = tmp_path / "string_view.parquet"
-
-        table = pa.table({"a": pa.array([None, "b", "c"], pa.string_view())})
-        pq.write_table(table, path)
-
-        with pd.option_context("future.infer_string", True):
-            result = read_parquet(path)
-        expected = pd.DataFrame(
-            data={"a": [None, "b", "c"]},
-            dtype=pd.StringDtype(na_value=np.nan),
-            columns=pd.Index(["a"], dtype=pd.StringDtype(na_value=np.nan)),
-        )
-        tm.assert_frame_equal(result, expected)
-
     # NOTE: this test is not run by default, because it requires a lot of memory (>5GB)
     # @pytest.mark.slow
     # def test_string_column_above_2GB(self, tmp_path, pa):