Add pytest gpu mark

2025-08-27 11:08:16 +08:00
parent 947a2a1583
commit 58721ca974
2 changed files with 7 additions and 7 deletions
--- a/tests/test_cuquantum_cutensor_backend.py
+++ b/tests/test_cuquantum_cutensor_backend.py
@@ -40,7 +40,7 @@ def build_observable_dict(nqubits):

    return {"terms": terms}

-
+@pytest.mark.gpu
@pytest.mark.parametrize("nqubits", [1, 2, 5, 10])
 def test_eval(nqubits: int, dtype="complex128"):
    """
@@ -81,7 +81,7 @@ def test_eval(nqubits: int, dtype="complex128"):
        result_sv_cp, result_tn.statevector.flatten()
    ), "Resulting dense vectors do not match"

-
+@pytest.mark.gpu
@pytest.mark.parametrize("nqubits", [2, 5, 10])
 def test_mps(nqubits: int, dtype="complex128"):
    """Evaluate MPS with cuQuantum.