tests: added single qubit parametrized tests

zapata-engineering · Oct 12, 2023 · eb27f38 · eb27f38
1 parent 500146f
commit eb27f38
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/orquestra/braket/simulator_test.py b/tests/orquestra/braket/simulator_test.py
@@ -33,7 +33,11 @@ def test_braket_local_wf_simulator_fulfills_circuit_runner_contracts(contract):
     assert contract(simulator)
 
 
-@pytest.mark.parametrize("contract", simulator_gate_compatibility_contracts())
+@pytest.mark.local
+# Braket doesn't support the RH gate, so we're excluding it from the test set.
+@pytest.mark.parametrize(
+    "contract", simulator_gate_compatibility_contracts(gates_to_exclude=["RH"])
+)
 def test_braket_simulator_uses_correct_gate_definitionscontract(contract):
     simulator = braket_local_simulator()
     assert contract(simulator)