From abfdbb02fbf37e986f986cdfefaf0c11063b980c Mon Sep 17 00:00:00 2001 From: Elliott Sales de Andrade Date: Sun, 18 Dec 2022 04:21:01 -0500 Subject: [PATCH] TST: Add missing skips for pyarrow --- distributed/shuffle/tests/test_shuffle_extension.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/distributed/shuffle/tests/test_shuffle_extension.py b/distributed/shuffle/tests/test_shuffle_extension.py index b40902b4f4c..1c106306166 100644 --- a/distributed/shuffle/tests/test_shuffle_extension.py +++ b/distributed/shuffle/tests/test_shuffle_extension.py @@ -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],