fix: make pyarrow optional (#33)

.github/workflows/pypi.yml

@@ -30,7 +30,7 @@ jobs:
     - name: install
       run: |
         pip install pip wheel -U
-        pip install -q --no-cache-dir .[complete]
+        pip install -q --no-cache-dir .[test]
         pip list
     - name: test
       run: |

pyproject.toml

@@ -35,15 +35,24 @@ dynamic = ["version"]
 
 [project.optional-dependencies]
 complete = [
-    "pyarrow",
-    "pytest >=6.0",
-    "pytest-cov >=3.0.0",
+  "awkward-pandas[strings,parquet]",
+]
+parquet = [
+  "pyarrow",
+]
+strings = [
+  "pyarrow >=7.0.0",
+]
+test = [
+  "awkward-pandas[complete]",
+  "pytest >=6.0",
+  "pytest-cov >=3.0.0",
 ]
 docs = [
-    "sphinx",
-    "furo",
-    "nbsphinx",
-    "ipython",
+  "sphinx",
+  "furo",
+  "nbsphinx",
+  "ipython",
 ]
 
 [project.urls]

tests/test_accessor.py

@@ -26,11 +26,14 @@ def test_getitem():
     assert s2.tolist() == [[6], [4]]
 
 
-def test_to_column():
+def test_to_column_ints():
     s = pd.Series(awkward_pandas.AwkwardExtensionArray([6, 2, 3]), name="test")
     s2 = s.ak.to_column()
     assert s2.dtype == "int64"
 
+
+def test_to_column_strings():
+    pytest.importorskip("pyarrow")
     s = pd.Series(awkward_pandas.AwkwardExtensionArray(["6", "2", "3"]), name="test")
     s2 = s.ak.to_column()
     assert s2.dtype == "string[pyarrow]"
@@ -41,6 +44,7 @@ def test_to_column():
 
 
 def test_to_columns():
+    pytest.importorskip("pyarrow")
     s = pd.Series(
         awkward_pandas.AwkwardExtensionArray(
             {"num": [6, 2, 3], "deep": [[0], [], None], "text": ["hi", "ho", "hum"]}

tests/test_array.py

@@ -1,12 +1,14 @@
 from __future__ import annotations
 
 import pandas as pd
+import pytest
 from pandas.testing import assert_frame_equal
 
 from awkward_pandas import AwkwardExtensionArray, merge
 
 
 def test_merge_no_ak():
+    pytest.importorskip("pyarrow")
     df = pd.DataFrame(
         {
             "a": [1, 2, 3],
@@ -40,6 +42,7 @@ def test_merge_one_ak():
 
 
 def test_parquet_roundtrip(tmp_path):
+    pytest.importorskip("pyarrow")
     df = pd.DataFrame(
         {
             "a": [1, 2, 3, 4, 5],

tests/test_str.py

@@ -3,6 +3,8 @@
 import pandas as pd
 import pytest
 
+pytest.importorskip("pyarrow")
+
 
 @pytest.mark.parametrize("binary", [True, False])
 @pytest.mark.parametrize("method", ["upper", "capitalize", "isalpha"])