def test_append_strategies():
a = cirq.QubitId()
b = cirq.QubitId()
stream = [cirq.X(a), cirq.CZ(a, b), cirq.X(b), cirq.X(b), cirq.X(a)]
c = Circuit()
c.append(stream, InsertStrategy.NEW)
assert c == Circuit([
Moment([cirq.X(a)]),
Moment([cirq.CZ(a, b)]),
Moment([cirq.X(b)]),
Moment([cirq.X(b)]),
Moment([cirq.X(a)]),
])
c = Circuit()
c.append(stream, InsertStrategy.INLINE)
assert c == Circuit([
Moment([cirq.X(a)]),
Moment([cirq.CZ(a, b)]),
Moment([cirq.X(b)]),
Moment([cirq.X(b), cirq.X(a)]),
])
c = Circuit()
c.append(stream, InsertStrategy.EARLIEST)
assert c == Circuit([
Moment([cirq.X(a)]),
Moment([cirq.CZ(a, b)]),
Moment([cirq.X(b), cirq.X(a)]),
Moment([cirq.X(b)]),
])
def test_append_single():
a = cirq.QubitId()
c = Circuit()
c.append(())
assert c == Circuit()
c = Circuit()
c.append(cirq.X(a))
assert c == Circuit([Moment([cirq.X(a)])])
c = Circuit()
c.append([cirq.X(a)])
assert c == Circuit([Moment([cirq.X(a)])])
def test_append_multiple():
a = cirq.QubitId()
b = cirq.QubitId()
c = Circuit()
c.append([cirq.X(a), cirq.X(b)], InsertStrategy.NEW)
assert c == Circuit([Moment([cirq.X(a)]), Moment([cirq.X(b)])])
c = Circuit()
c.append([cirq.X(a), cirq.X(b)], InsertStrategy.EARLIEST)
assert c == Circuit([
Moment([cirq.X(a), cirq.X(b)]),
])
c = Circuit()
c.append(cirq.X(a), InsertStrategy.EARLIEST)
c.append(cirq.X(b), InsertStrategy.EARLIEST)
assert c == Circuit([
Moment([cirq.X(a), cirq.X(b)]),
])
def assert_act_on_clifford_tableau_effect_matches_unitary(val: Any) -> None:
"""Checks that act_on with CliffordTableau generates stabilizers that
stabilize the final state vector. Does not work with Operations or Gates
expecting non-qubit Qids."""
# pylint: disable=unused-variable
__tracebackhide__ = True
# pylint: enable=unused-variable
num_qubits_val = protocols.num_qubits(val)
if not protocols.has_unitary(val) or \
protocols.qid_shape(val) != (2,) * num_qubits_val:
return None
qubits = LineQubit.range(protocols.num_qubits(val) * 2)
qubit_map = {qubit: i for i, qubit in enumerate(qubits)}
circuit = Circuit()
for i in range(num_qubits_val):
circuit.append([
common_gates.H(qubits[i]),
common_gates.CNOT(qubits[i], qubits[-i - 1])
])
if hasattr(val, "on"):
circuit.append(val.on(*qubits[:num_qubits_val]))
else:
circuit.append(val.with_qubits(*qubits[:num_qubits_val]))
tableau = _final_clifford_tableau(circuit, qubit_map)
if tableau is None:
return None
state_vector = np.reshape(final_state_vector(circuit, qubit_order=qubits),
protocols.qid_shape(qubits))
assert all(
state_vector_has_stabilizer(state_vector, stab)
for stab in tableau.stabilizers()), (
"act_on clifford tableau is not consistent with "
"final_state_vector simulation.\n\nval: {!r}".format(val))
def assert_all_implemented_act_on_effects_match_unitary(
val: Any,
assert_tableau_implemented: bool = False,
assert_ch_form_implemented: bool = False) -> None:
"""Uses val's effect on final_state_vector to check act_on(val)'s behavior.
Checks that act_on with CliffordTableau or StabilizerStateCHForm behaves
consistently with act_on through final state vector. Does not work with
Operations or Gates expecting non-qubit Qids. If either of the
assert_*_implmented args is true, fails if the corresponding method is not
implemented for the test circuit.
Args:
val: A gate or operation that may be an input to protocols.act_on.
assert_tableau_implemented: asserts that protocols.act_on() works with
val and ActOnCliffordTableauArgs inputs.
assert_ch_form_implemented: asserts that protocols.act_on() works with
val and ActOnStabilizerStateChFormArgs inputs.
"""
# pylint: disable=unused-variable
__tracebackhide__ = True
# pylint: enable=unused-variable
num_qubits_val = protocols.num_qubits(val)
if (protocols.is_parameterized(val) or not protocols.has_unitary(val)
or protocols.qid_shape(val) != (2, ) * num_qubits_val):
if assert_tableau_implemented or assert_ch_form_implemented:
assert False, ("Could not assert if any act_on methods were "
"implemented. Operating on qudits or with a "
"non-unitary or parameterized operation is "
"unsupported.\n\nval: {!r}".format(val))
return None
qubits = LineQubit.range(protocols.num_qubits(val) * 2)
qubit_map = {qubit: i for i, qubit in enumerate(qubits)}
circuit = Circuit()
for i in range(num_qubits_val):
circuit.append([
common_gates.H(qubits[i]),
common_gates.CNOT(qubits[i], qubits[-i - 1])
])
if hasattr(val, "on"):
circuit.append(val.on(*qubits[:num_qubits_val]))
else:
circuit.append(val.with_qubits(*qubits[:num_qubits_val]))
state_vector = np.reshape(final_state_vector(circuit, qubit_order=qubits),
protocols.qid_shape(qubits))
tableau = _final_clifford_tableau(circuit, qubit_map)
if tableau is None:
assert (
not assert_tableau_implemented
), "Failed to generate final tableau for the test circuit.\n\nval: {!r}".format(
val)
else:
assert all(
state_vector_has_stabilizer(state_vector, stab)
for stab in tableau.stabilizers()), (
"act_on clifford tableau is not consistent with "
"final_state_vector simulation.\n\nval: {!r}".format(val))
stabilizer_ch_form = _final_stabilizer_state_ch_form(circuit, qubit_map)
if stabilizer_ch_form is None:
assert not assert_ch_form_implemented, ("Failed to generate final "
"stabilizer state CH form "
"for the test circuit."
"\n\nval: {!r}".format(val))
else:
np.testing.assert_allclose(
np.reshape(stabilizer_ch_form.state_vector(),
protocols.qid_shape(qubits)),
state_vector,
atol=1e-07,
err_msg=
f"stabilizer_ch_form.state_vector disagrees with state_vector for {val!r}",
verbose=True,
)
