def test_cphase_gates():
for _ in range(REPS):
theta = random.uniform(-4 * pi, +4 * pi)
gate11 = qf.control_gate(0, qf.PHASE(theta, 1))
assert qf.gates_close(gate11, qf.CPHASE(theta, 0, 1))
gate01 = qf.conditional_gate(0, qf.PHASE(theta, 1), qf.I(1))
assert qf.gates_close(gate01, qf.CPHASE01(theta))
gate00 = qf.identity_gate(2)
gate00 = qf.X(0) @ gate00
gate00 = qf.X(1) @ gate00
gate00 = gate11 @ gate00
gate00 = qf.X(0) @ gate00
gate00 = qf.X(1) @ gate00
assert qf.gates_close(gate00, qf.CPHASE00(theta))
gate10 = qf.identity_gate(2)
gate10 = qf.X(0) @ gate10
gate10 = qf.X(1) @ gate10
gate10 = gate01 @ gate10
gate10 = qf.X(0) @ gate10
gate10 = qf.X(1) @ gate10
assert qf.gates_close(gate10, qf.CPHASE10(theta))
def test_control_gate():
assert qf.gates_close(qf.control_gate(1, qf.X(0)), qf.CNOT(1, 0))
assert qf.gates_close(qf.control_gate(0, qf.CNOT(1, 2)), qf.CCNOT(0, 1, 2))