From 6693e93bec6f36c6c849b0651810a9306e2580f3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ra=C3=BAl=20Cumplido?= Date: Wed, 26 Jun 2024 17:55:03 +0200 Subject: [PATCH] Add pytest.mark.numpy to all test_cuda and test_cuda_numba_interop tests --- python/pyarrow/tests/test_cuda.py | 5 ++++- python/pyarrow/tests/test_cuda_numba_interop.py | 5 ++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/python/pyarrow/tests/test_cuda.py b/python/pyarrow/tests/test_cuda.py index 5e833d5371de2..a49fea7f88ec5 100644 --- a/python/pyarrow/tests/test_cuda.py +++ b/python/pyarrow/tests/test_cuda.py @@ -26,7 +26,10 @@ import pytest import pyarrow as pa -import numpy as np +try: + import numpy as np +except ImportError: + pytestmark = pytest.mark.numpy cuda = pytest.importorskip("pyarrow.cuda") diff --git a/python/pyarrow/tests/test_cuda_numba_interop.py b/python/pyarrow/tests/test_cuda_numba_interop.py index ff1722d278d5e..876f3c7f761cf 100644 --- a/python/pyarrow/tests/test_cuda_numba_interop.py +++ b/python/pyarrow/tests/test_cuda_numba_interop.py @@ -17,7 +17,10 @@ import pytest import pyarrow as pa -import numpy as np +try: + import numpy as np +except ImportError: + pytestmark = pytest.mark.numpy dtypes = ['uint8', 'int16', 'float32'] cuda = pytest.importorskip("pyarrow.cuda")