Add missing skips for pyarrow
As in test_split_by_partition in the same file.
@@ -23,6 +23,8 @@ async def test_installation(s, a):
def test_split_by_worker():
+ pytest.importorskip("pyarrow")
+
df = pd.DataFrame(
{
"x": [1, 2, 3, 4, 5],
@@ -44,6 +46,8 @@ def test_split_by_worker():
def test_split_by_worker_empty():
+ pytest.importorskip("pyarrow")
+
df = pd.DataFrame(
{
"x": [1, 2, 3, 4, 5],
@@ -56,6 +60,8 @@ def test_split_by_worker_empty():
def test_split_by_worker_many_workers():
+ pytest.importorskip("pyarrow")
+
df = pd.DataFrame(
{
"x": [1, 2, 3, 4, 5],
