Update test_controlled_gate.py

comp-quantica · Feb 29, 2024 · d811feb · d811feb
1 parent 524a5b3
commit d811feb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/python/circuit/test_controlled_gate.py b/test/python/circuit/test_controlled_gate.py
@@ -802,10 +802,11 @@ def test_mcxvchain_dirty_ancilla_cx_count(self, num_ctrl_qubits):
         self.assertLessEqual(cx_count, 8 * num_ctrl_qubits - 6)
 
     def test_mcxvchain_dirty_ancilla_action_only(self):
-        num_ctrl_qubits = 5
         """Test the v-chain mcx with dirty auxiliary qubits
         with gate cancelling with mirrored circuit."""
 
+        num_ctrl_qubits = 5
+
         gate = MCXVChain(num_ctrl_qubits, dirty_ancillas=True)
         gate_with_cancelling = MCXVChain(num_ctrl_qubits, dirty_ancillas=True, action_only=True)
 
@@ -824,9 +825,9 @@ def test_mcxvchain_dirty_ancilla_action_only(self):
         self.assertTrue(matrix_equal(Operator(circuit).data, Operator(ref_circuit).data))
 
     def test_mcxvchain_dirty_ancilla_relative_phase(self):
-        num_ctrl_qubits = 5
         """Test the v-chain mcx with dirty auxiliary qubits
         with only relative phase Toffoli gates."""
+        num_ctrl_qubits = 5
 
         gate = MCXVChain(num_ctrl_qubits, dirty_ancillas=True)
         gate_relative_phase = MCXVChain(num_ctrl_qubits, dirty_ancillas=True, relative_phase=True)