fix fixed-qubit test

rigetti · May 23, 2024 · f492c36 · f492c36
1 parent 9ac6ad1
commit f492c36
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/test/unit/__snapshots__/test_quilbase.ambr b/test/unit/__snapshots__/test_quilbase.ambr
@@ -138,6 +138,9 @@
 # name: TestDeclare.test_str[With-Size]
   'DECLARE ro OCTET[5]'
 # ---
+# name: TestDefCalibration.test_out[Fixed-Qubit]
+  'DEFCAL Calibrate 0:'
+# ---
 # name: TestDefCalibration.test_out[No-Params]
   '''
   DEFCAL Calibrate 0:
@@ -150,6 +153,9 @@
   	X 0
   '''
 # ---
+# name: TestDefCalibration.test_str[Fixed-Qubit]
+  'DEFCAL Calibrate 0:'
+# ---
 # name: TestDefCalibration.test_str[No-Params]
   '''
   DEFCAL Calibrate 0:

diff --git a/test/unit/test_quilbase.py b/test/unit/test_quilbase.py
@@ -412,7 +412,9 @@ def test_parameters(self, calibration: DefCalibration, parameters: List[Paramete
         calibration.parameters = [pi / 2]
         assert calibration.parameters == [pi / 2]
 
-    def test_qubits(self, calibration: DefCalibration, qubits: List[Qubit]):
+    def test_qubits(self, calibration: DefCalibration, qubits: List[QubitDesignator]):
+        if len(qubits) > 0 and isinstance(qubits[0], int):
+            qubits = [Qubit(index) for index in qubits]
         assert calibration.qubits == qubits
         calibration.qubits = [Qubit(123)]
         assert calibration.qubits == [Qubit(123)]