mirror of https://github.com/Qiskit/qiskit.git
93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
# This code is part of Qiskit.
|
|
#
|
|
# (C) Copyright IBM 2019, 2023.
|
|
#
|
|
# This code is licensed under the Apache License, Version 2.0. You may
|
|
# obtain a copy of this license in the LICENSE.txt file in the root directory
|
|
# of this source tree or at http://www.apache.org/licenses/LICENSE-2.0.
|
|
#
|
|
# Any modifications or derivative works of this code must retain this
|
|
# copyright notice, and modified files need to carry a notice indicating
|
|
# that they have been altered from the originals.
|
|
|
|
"""Tests for uniformly controlled Rx,Ry and Rz gates"""
|
|
|
|
import itertools
|
|
import unittest
|
|
import numpy as np
|
|
from scipy.linalg import block_diag
|
|
|
|
from qiskit import QuantumCircuit, QuantumRegister
|
|
from qiskit.quantum_info.operators.predicates import matrix_equal
|
|
from qiskit.quantum_info import Operator
|
|
from qiskit.compiler import transpile
|
|
from qiskit.circuit.library import UCRXGate, UCRYGate, UCRZGate
|
|
from test import QiskitTestCase # pylint: disable=wrong-import-order
|
|
|
|
angles_list = [
|
|
[0],
|
|
[0.4],
|
|
[0, 0],
|
|
[0, 0.8],
|
|
[0, 0, 1, 1],
|
|
[0, 1, 0.5, 1],
|
|
(2 * np.pi * np.random.rand(2**3)).tolist(),
|
|
(2 * np.pi * np.random.rand(2**4)).tolist(),
|
|
(2 * np.pi * np.random.rand(2**5)).tolist(),
|
|
]
|
|
|
|
rot_axis_list = ["X", "Y", "Z"]
|
|
|
|
|
|
class TestUCRXYZ(QiskitTestCase):
|
|
"""Qiskit tests for UCRXGate, UCRYGate and UCRZGate rotations gates."""
|
|
|
|
def test_ucy(self):
|
|
"""Test the decomposition of uniformly controlled rotations."""
|
|
gates = {"X": UCRXGate, "Y": UCRYGate, "Z": UCRZGate}
|
|
|
|
for angles, rot_axis in itertools.product(angles_list, rot_axis_list):
|
|
with self.subTest(angles=angles, rot_axis=rot_axis):
|
|
num_contr = int(np.log2(len(angles)))
|
|
q = QuantumRegister(num_contr + 1)
|
|
qc = QuantumCircuit(q)
|
|
gate = gates[rot_axis](angles)
|
|
qc.append(gate, q)
|
|
|
|
# Decompose the gate
|
|
qc = transpile(qc, basis_gates=["u1", "u3", "u2", "cx", "id"])
|
|
# Simulate the decomposed gate
|
|
unitary = Operator(qc)
|
|
unitary_desired = _get_ucr_matrix(angles, rot_axis)
|
|
self.assertTrue(matrix_equal(unitary_desired, unitary, ignore_phase=True))
|
|
|
|
|
|
def _get_ucr_matrix(angles, rot_axis):
|
|
if rot_axis == "X":
|
|
gates = [
|
|
np.array(
|
|
[
|
|
[np.cos(angle / 2), -1j * np.sin(angle / 2)],
|
|
[-1j * np.sin(angle / 2), np.cos(angle / 2)],
|
|
]
|
|
)
|
|
for angle in angles
|
|
]
|
|
elif rot_axis == "Y":
|
|
gates = [
|
|
np.array(
|
|
[[np.cos(angle / 2), -np.sin(angle / 2)], [np.sin(angle / 2), np.cos(angle / 2)]]
|
|
)
|
|
for angle in angles
|
|
]
|
|
else:
|
|
gates = [
|
|
np.array([[np.exp(-1.0j * angle / 2), 0], [0, np.exp(1.0j * angle / 2)]])
|
|
for angle in angles
|
|
]
|
|
return block_diag(*gates)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|