install cuda via pyproject; add cupy test

.github/workflows/ci-gpu.yml

@@ -42,8 +42,7 @@ jobs:
         run: |
           nvidia-smi
           python3 -m pip install --upgrade pip
-          pip3 install cupy-cuda114
-          pip install -e .[test]
+          pip install -e .[test-cuda114]
           pytest test
         env:
           REPO_TOKEN: ${{ secrets.PERSONAL_ACCESS_TOKEN }}

pyproject.toml

@@ -44,6 +44,11 @@ test = [
     'pytest==7.4.*',
     'invoke==2.2.*',
     ]
+test-cuda114 = [
+    'cupy-cuda114==10.6.*',
+    'pytest==7.4.*',
+    'invoke==2.2.*',
+    ]
 
 # [project.scripts]
 # my-script = "my_package.module:function"

test/unit/test_curay.py

@@ -4,8 +4,17 @@
 from hyray import ndcuray
 import hyray as hr
 
+# NOTE: these tests require CUDA
+
 # pytest -W ignore::DeprecationWarning test/test_unit.py
 
+
+def test_cupy():
+    # this test is to proove that cupy is intalled and working
+    import cupy as cp
+    print('cupy version:', cp.__version__)
+    a1 = cp.array([0, 1])
+
 #-------------------------------------------------------------------------------
 # test ndcurayFlags object
 

test/unit/test_hyray.py

@@ -2,6 +2,8 @@
 import hyray as hr
 import numpy as np
 
+# NOTE: this test can be run (and should pass) without CUDA
+
 MODULES = (hr, np)
 
 @pytest.mark.parametrize('mod', MODULES)