def test_parametrize(self):
"""Tests that parametrize yields the correct queue of operations."""
operation = CirqOperation(
lambda a, b, c:
[cirq.X, cirq.ry(a),
cirq.rx(b), cirq.Z,
cirq.rz(c)])
operation.parametrize(0.1, 0.2, 0.3)
assert operation.parametrized_cirq_gates[0] == cirq.X
assert operation.parametrized_cirq_gates[1] == cirq.ry(0.1)
assert operation.parametrized_cirq_gates[2] == cirq.rx(0.2)
assert operation.parametrized_cirq_gates[3] == cirq.Z
assert operation.parametrized_cirq_gates[4] == cirq.rz(0.3)
def test_apply_not_parametrized(self):
"""Tests that the proper error is raised if an Operation is applied
that was not parametrized before."""
operation = CirqOperation(
lambda a, b, c:
[cirq.X, cirq.ry(a),
cirq.rx(b), cirq.Z,
cirq.rz(c)])
qubit = cirq.LineQubit(1)
with pytest.raises(
qml.DeviceError,
match=
"CirqOperation must be parametrized before it can be applied.",
):
operation.apply(qubit)
def test_init(self):
"""Tests that the class is properly initialized."""
fun = lambda x: cirq.ry(x)
operation = CirqOperation(fun)
assert operation.parametrized_cirq_gates is None
assert operation.parametrization == fun
assert not operation.is_inverse
def test_apply(self):
"""Tests that the operations in the queue are correctly applied."""
operation = CirqOperation(
lambda a, b, c:
[cirq.X, cirq.ry(a),
cirq.rx(b), cirq.Z,
cirq.rz(c)])
operation.parametrize(0.1, 0.2, 0.3)
qubit = cirq.LineQubit(1)
gate_applications = list(operation.apply(qubit))
assert gate_applications[0] == cirq.X.on(qubit)
assert gate_applications[1] == cirq.ry(0.1).on(qubit)
assert gate_applications[2] == cirq.rx(0.2).on(qubit)
assert gate_applications[3] == cirq.Z.on(qubit)
assert gate_applications[4] == cirq.rz(0.3).on(qubit)
def test_inv_error(self):
"""Test that inv raises an error if the CirqOperation was already parametrized."""
operation = CirqOperation(
lambda a, b, c:
[cirq.X, cirq.ry(a),
cirq.rx(b), cirq.Z,
cirq.rz(c)])
operation.parametrize(0.1, 0.2, 0.3)
with pytest.raises(
qml.DeviceError,
match=
"CirqOperation can't be inverted after it was parametrized",
):
operation.inv()
def test_inv(self):
"""Test that inv inverts the gate and applying inv twice yields the initial state."""
operation = CirqOperation(
lambda a, b, c:
[cirq.X, cirq.ry(a),
cirq.rx(b), cirq.Z,
cirq.rz(c)])
assert not operation.is_inverse
operation.inv()
assert operation.is_inverse
operation.inv()
assert not operation.is_inverse