def test_deserialize_circuit_with_token_strings():
"""Supporting token strings for backwards compatibility."""
serializer = cg.CircuitSerializer('my_gate_set')
proto = v2.program_pb2.Program(
language=v2.program_pb2.Language(arg_function_language='exp',
gate_set='my_gate_set'),
circuit=v2.program_pb2.Circuit(
scheduling_strategy=v2.program_pb2.Circuit.MOMENT_BY_MOMENT,
moments=[
v2.program_pb2.Moment(operations=[
v2.program_pb2.Operation(
xpowgate=v2.program_pb2.XPowGate(
exponent=v2.program_pb2.FloatArg(float_value=1.0)),
token_value='abc123',
qubit_constant_index=[0],
),
], ),
],
),
constants=[
v2.program_pb2.Constant(qubit=v2.program_pb2.Qubit(id='2_4')),
],
)
tag = cg.CalibrationTag('abc123')
circuit = cirq.Circuit(cirq.X(Q0).with_tags(tag))
assert serializer.deserialize(proto) == circuit
def test_deserialize_unsupported_gate_type():
serializer = cg.CircuitSerializer('my_gate_set')
operation_proto = op_proto({
'gate': {
'id': 'no_pow'
},
'args': {
'half_turns': {
'arg_value': {
'float_value': 0.125
}
},
},
'qubits': [{
'id': '1_1'
}],
})
proto = v2.program_pb2.Program(
language=v2.program_pb2.Language(arg_function_language='',
gate_set='my_gate_set'),
circuit=v2.program_pb2.Circuit(
scheduling_strategy=v2.program_pb2.Circuit.MOMENT_BY_MOMENT,
moments=[
v2.program_pb2.Moment(operations=[operation_proto]),
],
),
)
with pytest.raises(ValueError, match='no_pow'):
serializer.deserialize(proto)
def test_deserialize_wrong_types():
serializer = cg.CircuitSerializer('my_gate_set')
proto = circuit_proto(
{
'measurementgate': {
'key': {
'arg_value': {
'float_value': 3.0
}
},
'invert_mask': {
'arg_value': {
'bool_values': {
'values': [True, False]
}
}
},
},
'qubit_constant_index': [0],
},
['1_1'],
)
with pytest.raises(ValueError,
match='Incorrect types for measurement gate'):
serializer.deserialize(proto)
def test_deserialize_schedule_not_supported():
serializer = cg.CircuitSerializer('my_gate_set')
proto = v2.program_pb2.Program(
language=v2.program_pb2.Language(gate_set='my_gate_set'),
schedule=v2.program_pb2.Schedule(scheduled_operations=[
v2.program_pb2.ScheduledOperation(start_time_picos=0)
]),
)
with pytest.raises(ValueError, match='no longer supported'):
serializer.deserialize(proto, cg.Bristlecone)
def test_no_constants_table():
serializer = cg.CircuitSerializer('my_gate_set')
op = op_proto(
{
'xpowgate': {'exponent': {'float_value': 1.0}},
'qubits': [{'id': '1_2'}],
'token_constant_index': 0,
}
)
with pytest.raises(ValueError, match='Proto has references to constants table'):
serializer._deserialize_gate_op(op)
def test_serialize_deserialize_empty_circuit():
serializer = cg.CircuitSerializer('my_gate_set')
circuit = cirq.Circuit()
proto = v2.program_pb2.Program(
language=v2.program_pb2.Language(arg_function_language='exp', gate_set='my_gate_set'),
circuit=v2.program_pb2.Circuit(
scheduling_strategy=v2.program_pb2.Circuit.MOMENT_BY_MOMENT, moments=[]
),
)
assert proto == serializer.serialize(circuit)
assert serializer.deserialize(proto) == circuit
def test_serialize_deserialize_circuit_with_tokens():
serializer = cg.CircuitSerializer('my_gate_set')
tag1 = cg.CalibrationTag('abc123')
tag2 = cg.CalibrationTag('def456')
circuit = cirq.Circuit(
cirq.X(Q0).with_tags(tag1),
cirq.X(Q1).with_tags(tag2),
cirq.X(Q0).with_tags(tag2),
cirq.X(Q0),
)
op_q0_tag1 = v2.program_pb2.Operation()
op_q0_tag1.xpowgate.exponent.float_value = 1.0
op_q0_tag1.qubit_constant_index.append(0)
op_q0_tag1.token_constant_index = 1
op_q1_tag2 = v2.program_pb2.Operation()
op_q1_tag2.xpowgate.exponent.float_value = 1.0
op_q1_tag2.qubit_constant_index.append(2)
op_q1_tag2.token_constant_index = 3
# Test repeated tag uses existing constant entey
op_q0_tag2 = v2.program_pb2.Operation()
op_q0_tag2.xpowgate.exponent.float_value = 1.0
op_q0_tag2.qubit_constant_index.append(0)
op_q0_tag2.token_constant_index = 3
proto = v2.program_pb2.Program(
language=v2.program_pb2.Language(arg_function_language='exp', gate_set='my_gate_set'),
circuit=v2.program_pb2.Circuit(
scheduling_strategy=v2.program_pb2.Circuit.MOMENT_BY_MOMENT,
moments=[
v2.program_pb2.Moment(
operations=[op_q0_tag1, op_q1_tag2],
),
v2.program_pb2.Moment(
operations=[op_q0_tag2],
),
v2.program_pb2.Moment(
operations=[X_PROTO],
),
],
),
constants=[
v2.program_pb2.Constant(qubit=v2.program_pb2.Qubit(id='2_4')),
v2.program_pb2.Constant(string_value='abc123'),
v2.program_pb2.Constant(qubit=v2.program_pb2.Qubit(id='2_5')),
v2.program_pb2.Constant(string_value='def456'),
],
)
assert proto == serializer.serialize(circuit)
assert serializer.deserialize(proto) == circuit
def test_deserialize_fsim_missing_parameters():
serializer = cg.CircuitSerializer('my_gate_set')
proto = circuit_proto(
{
'fsimgate': {
'theta': {'float_value': 3.0},
},
'qubit_constant_index': [0, 1],
},
['1_1', '1_2'],
)
with pytest.raises(ValueError, match='theta and phi must be specified'):
serializer.deserialize(proto)
def test_serialize_op_bad_operation():
serializer = cg.CircuitSerializer('my_gate_set')
class NullOperation(cirq.Operation):
@property
def qubits(self):
return tuple() # coverage: ignore
def with_qubits(self, *qubits):
return self # coverage: ignore
null_op = NullOperation()
with pytest.raises(ValueError, match='Cannot serialize op'):
serializer.serialize(cirq.Circuit(null_op))
def test_serialize_circuit_op_errors():
serializer = cg.CircuitSerializer('my_gate_set')
constants = [default_circuit_proto()]
raw_constants = {default_circuit(): 0}
op = cirq.CircuitOperation(default_circuit())
with pytest.raises(ValueError, match='CircuitOp serialization requires a constants list'):
serializer._serialize_circuit_op(op)
with pytest.raises(ValueError, match='CircuitOp serialization requires a constants list'):
serializer._serialize_circuit_op(op, constants=constants)
with pytest.raises(ValueError, match='CircuitOp serialization requires a constants list'):
serializer._serialize_circuit_op(op, raw_constants=raw_constants)
def test_serialize_deserialize_circuit():
serializer = cg.CircuitSerializer('my_gate_set')
q0 = cirq.GridQubit(1, 1)
q1 = cirq.GridQubit(1, 2)
circuit = cirq.Circuit(cirq.X(q0), cirq.X(q1), cirq.X(q0))
proto = v2.program_pb2.Program(
language=v2.program_pb2.Language(arg_function_language='exp', gate_set='my_gate_set'),
circuit=v2.program_pb2.Circuit(
scheduling_strategy=v2.program_pb2.Circuit.MOMENT_BY_MOMENT,
moments=[
v2.program_pb2.Moment(
operations=[
v2.program_pb2.Operation(
xpowgate=v2.program_pb2.XPowGate(
exponent=v2.program_pb2.FloatArg(float_value=1.0)
),
qubit_constant_index=[0],
),
v2.program_pb2.Operation(
xpowgate=v2.program_pb2.XPowGate(
exponent=v2.program_pb2.FloatArg(float_value=1.0)
),
qubit_constant_index=[1],
),
],
),
v2.program_pb2.Moment(
operations=[
v2.program_pb2.Operation(
xpowgate=v2.program_pb2.XPowGate(
exponent=v2.program_pb2.FloatArg(float_value=1.0)
),
qubit_constant_index=[0],
),
],
),
],
),
constants=[
v2.program_pb2.Constant(qubit=v2.program_pb2.Qubit(id='1_1')),
v2.program_pb2.Constant(qubit=v2.program_pb2.Qubit(id='1_2')),
],
)
assert proto == serializer.serialize(circuit)
assert serializer.deserialize(proto) == circuit
def test_serialize_deserialize_ops(op, op_proto):
serializer = cg.CircuitSerializer('my_gate_set')
constants = []
for q in op.qubits:
constants.append(v2.program_pb2.Constant(qubit=v2.program_pb2.Qubit(id=f'{q.row}_{q.col}')))
# Serialize / Deserializer circuit with single operation
circuit = cirq.Circuit(op)
circuit_proto = v2.program_pb2.Program(
language=v2.program_pb2.Language(arg_function_language='exp', gate_set='my_gate_set'),
circuit=v2.program_pb2.Circuit(
scheduling_strategy=v2.program_pb2.Circuit.MOMENT_BY_MOMENT,
moments=[v2.program_pb2.Moment(operations=[op_proto])],
),
constants=constants,
)
assert circuit_proto == serializer.serialize(circuit)
assert serializer.deserialize(circuit_proto) == circuit
def test_deserialize_invalid_gate_set():
serializer = cg.CircuitSerializer('my_gate_set')
proto = v2.program_pb2.Program(
language=v2.program_pb2.Language(gate_set='not_my_gate_set'),
circuit=v2.program_pb2.Circuit(
scheduling_strategy=v2.program_pb2.Circuit.MOMENT_BY_MOMENT,
moments=[]),
)
with pytest.raises(ValueError, match='not_my_gate_set'):
serializer.deserialize(proto)
proto.language.gate_set = ''
with pytest.raises(ValueError, match='Missing gate set'):
serializer.deserialize(proto)
proto = v2.program_pb2.Program(circuit=v2.program_pb2.Circuit(
scheduling_strategy=v2.program_pb2.Circuit.MOMENT_BY_MOMENT,
moments=[]))
with pytest.raises(ValueError, match='Missing gate set'):
serializer.deserialize(proto)
def test_serialize_deserialize_circuit_with_subcircuit():
serializer = cg.CircuitSerializer('my_gate_set')
tag1 = cg.CalibrationTag('abc123')
fcircuit = cirq.FrozenCircuit(
cirq.XPowGate(exponent=2 * sympy.Symbol('t'))(Q0))
circuit = cirq.Circuit(
cirq.X(Q1).with_tags(tag1),
cirq.CircuitOperation(fcircuit).repeat(repetition_ids=['a', 'b']),
cirq.CircuitOperation(fcircuit).with_qubit_mapping({Q0: Q1}),
cirq.X(Q0),
)
op_x = v2.program_pb2.Operation()
op_x.xpowgate.exponent.float_value = 1.0
op_x.qubit_constant_index.append(2)
op_tag = v2.program_pb2.Operation()
op_tag.xpowgate.exponent.float_value = 1.0
op_tag.qubit_constant_index.append(0)
op_tag.token_constant_index = 1
op_symbol = v2.program_pb2.Operation()
op_symbol.xpowgate.exponent.func.type = 'mul'
op_symbol.xpowgate.exponent.func.args.add().arg_value.float_value = 2.0
op_symbol.xpowgate.exponent.func.args.add().symbol = 't'
op_symbol.qubit_constant_index.append(2)
c_op1 = v2.program_pb2.CircuitOperation()
c_op1.circuit_constant_index = 3
rep_spec = c_op1.repetition_specification
rep_spec.repetition_count = 2
rep_spec.repetition_ids.ids.extend(['a', 'b'])
c_op2 = v2.program_pb2.CircuitOperation()
c_op2.circuit_constant_index = 3
c_op2.repetition_specification.repetition_count = 1
qmap = c_op2.qubit_map.entries.add()
qmap.key.id = '2_4'
qmap.value.id = '2_5'
proto = v2.program_pb2.Program(
language=v2.program_pb2.Language(arg_function_language='exp',
gate_set='my_gate_set'),
circuit=v2.program_pb2.Circuit(
scheduling_strategy=v2.program_pb2.Circuit.MOMENT_BY_MOMENT,
moments=[
v2.program_pb2.Moment(
operations=[op_tag],
circuit_operations=[c_op1],
),
v2.program_pb2.Moment(
operations=[op_x],
circuit_operations=[c_op2],
),
],
),
constants=[
v2.program_pb2.Constant(qubit=v2.program_pb2.Qubit(id='2_5')),
v2.program_pb2.Constant(string_value='abc123'),
v2.program_pb2.Constant(qubit=v2.program_pb2.Qubit(id='2_4')),
v2.program_pb2.Constant(circuit_value=v2.program_pb2.Circuit(
scheduling_strategy=v2.program_pb2.Circuit.MOMENT_BY_MOMENT,
moments=[v2.program_pb2.Moment(operations=[op_symbol], )],
)),
],
)
assert proto == serializer.serialize(circuit)
assert serializer.deserialize(proto) == circuit
def test_circuit_serializer_name():
serializer = cg.CircuitSerializer('my_gate_set')
assert serializer.name == 'my_gate_set'
serializer = cg.CircuitSerializer('serial_box')
assert serializer.name == 'serial_box'
assert cg.serialization.circuit_serializer.CIRCUIT_SERIALIZER.name == 'v2_5'
def test_serialize_unrecognized():
serializer = cg.CircuitSerializer('my_gate_set')
with pytest.raises(NotImplementedError, match='program type'):
serializer.serialize("not quite right")
def test_serialize_op_unsupported_type():
serializer = cg.CircuitSerializer('my_gate_set')
q0 = cirq.GridQubit(1, 1)
q1 = cirq.GridQubit(1, 2)
with pytest.raises(ValueError, match='CNOT'):
serializer.serialize(cirq.Circuit(cirq.CNOT(q0, q1)))