Merge pull request #206 from qiboteam/gpu_only_selfhosted

Option to run only gpu tests in conftest

selfhosted

@@ -11,7 +11,8 @@ cd "$TMP_DIR/tests"
 
 source /nfs/users/github/actions-runner/_work/qibojit/qibojit/testenv/bin/activate
 
-pytest
+pytest --gpu-only
+
 pytest_status=$?
 
 if [[ $pytest_status -ne 0 ]]

src/qibojit/tests/conftest.py

@@ -2,6 +2,16 @@
 
 from qibojit.backends import CupyBackend, CuQuantumBackend, NumbaBackend
 
+
+def pytest_addoption(parser):
+    parser.addoption(
+        "--gpu-only",
+        action="store_true",
+        default=False,
+        help="Run on GPU backends only.",
+    )
+
+
 BACKENDS = {"numba": NumbaBackend, "cupy": CupyBackend, "cuquantum": CuQuantumBackend}
 
 # ignore backends that are not available in the current testing environment
@@ -15,7 +25,10 @@
 
 
 @pytest.fixture
-def backend(backend_name):
+def backend(backend_name, request):
+    if request.config.getoption("--gpu-only"):  # pragma: no cover
+        if backend_name not in ("cupy", "cuquantum"):
+            pytest.skip("Skipping non-gpu backend.")
     yield BACKENDS.get(backend_name)()