def test_floquet_get_calibrations() -> None: parameters_ab = cirq_google.PhasedFSimCharacterization( theta=0.6, zeta=0.5, chi=0.4, gamma=0.3, phi=0.2 ) parameters_bc = cirq_google.PhasedFSimCharacterization( theta=0.8, zeta=-0.5, chi=-0.4, gamma=-0.3, phi=-0.2 ) parameters_cd_dict = {'theta': 0.1, 'zeta': 0.2, 'chi': 0.3, 'gamma': 0.4, 'phi': 0.5} parameters_cd = cirq_google.PhasedFSimCharacterization(**parameters_cd_dict) a, b, c, d = cirq.LineQubit.range(4) engine_simulator = PhasedFSimEngineSimulator.create_from_dictionary_sqrt_iswap( parameters={(a, b): parameters_ab, (b, c): parameters_bc, (c, d): parameters_cd_dict} ) requests = [_create_sqrt_iswap_request([(a, b), (c, d)]), _create_sqrt_iswap_request([(b, c)])] results = engine_simulator.get_calibrations(requests) assert results == [ cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, 0.0), parameters={(a, b): parameters_ab, (c, d): parameters_cd}, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ), cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, 0.0), parameters={(b, c): parameters_bc}, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ), ]
def test_from_characterizations_sqrt_iswap_simulates_correctly(): parameters_ab = cirq_google.PhasedFSimCharacterization(theta=0.6, zeta=0.5, chi=0.4, gamma=0.3, phi=0.2) parameters_bc = cirq_google.PhasedFSimCharacterization(theta=0.8, zeta=-0.5, chi=-0.4, gamma=-0.3, phi=-0.2) parameters_cd = cirq_google.PhasedFSimCharacterization(theta=0.1, zeta=0.2, chi=0.3, gamma=0.4, phi=0.5) a, b, c, d = cirq.LineQubit.range(4) circuit = cirq.Circuit([ [cirq.X(a), cirq.Y(c)], [ cirq.FSimGate(np.pi / 4, 0.0).on(a, b), cirq.FSimGate(np.pi / 4, 0.0).on(c, d) ], [cirq.FSimGate(np.pi / 4, 0.0).on(c, b)], ]) expected_circuit = cirq.Circuit([ [cirq.X(a), cirq.X(c)], [ cirq.PhasedFSimGate(**parameters_ab.asdict()).on(a, b), cirq.PhasedFSimGate(**parameters_cd.asdict()).on(c, d), ], [cirq.PhasedFSimGate(**parameters_bc.asdict()).on(b, c)], ]) engine_simulator = PhasedFSimEngineSimulator.create_from_characterizations_sqrt_iswap( characterizations=[ cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, 0.0), parameters={ (a, b): parameters_ab, (c, d): parameters_cd }, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ), cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, 0.0), parameters={ (c, b): parameters_bc.parameters_for_qubits_swapped() }, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ), ]) actual = engine_simulator.final_state_vector(circuit) expected = cirq.final_state_vector(expected_circuit) assert cirq.allclose_up_to_global_phase(actual, expected)
def test_from_characterizations_sqrt_iswap_when_invalid_arguments_fails(): parameters_ab = cirq_google.PhasedFSimCharacterization(theta=0.6, zeta=0.5, chi=0.4, gamma=0.3, phi=0.2) parameters_bc = cirq_google.PhasedFSimCharacterization(theta=0.8, zeta=-0.5, chi=-0.4, gamma=-0.3, phi=-0.2) a, b = cirq.LineQubit.range(2) with pytest.raises(ValueError, match="multiple moments"): PhasedFSimEngineSimulator.create_from_characterizations_sqrt_iswap( characterizations=[ cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, 0.0), parameters={(a, b): parameters_ab}, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ), cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, 0.0), parameters={(a, b): parameters_bc}, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ), ]) with pytest.raises(AssertionError, match="Expected ISWA"): PhasedFSimEngineSimulator.create_from_characterizations_sqrt_iswap( characterizations=[ cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, 0.2), parameters={(a, b): parameters_ab}, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ) ]) with pytest.raises(ValueError, match="unparameterized"): PhasedFSimEngineSimulator.create_from_characterizations_sqrt_iswap( characterizations=[ cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, sympy.Symbol("a")), parameters={(a, b): parameters_ab}, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ) ]) with pytest.raises(AssertionError, match="Expected FSimGate"): PhasedFSimEngineSimulator.create_from_characterizations_sqrt_iswap( characterizations=[ cirq_google.PhasedFSimCalibrationResult( gate=cirq.CNOT, parameters={(a, b): parameters_ab}, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ) ])
def test_from_characterizations_sqrt_iswap_when_invalid_arguments_fails() -> None: parameters_ab = cirq_google.PhasedFSimCharacterization( theta=0.6, zeta=0.5, chi=0.4, gamma=0.3, phi=0.2 ) parameters_bc = cirq_google.PhasedFSimCharacterization( theta=0.8, zeta=-0.5, chi=-0.4, gamma=-0.3, phi=-0.2 ) a, b = cirq.LineQubit.range(2) with pytest.raises(ValueError): PhasedFSimEngineSimulator.create_from_characterizations_sqrt_iswap( characterizations=[ cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, 0.0), parameters={(a, b): parameters_ab}, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ), cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, 0.0), parameters={(a, b): parameters_bc}, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ), ] ) with pytest.raises(ValueError): PhasedFSimEngineSimulator.create_from_characterizations_sqrt_iswap( characterizations=[ cirq_google.PhasedFSimCalibrationResult( gate=cirq.FSimGate(np.pi / 4, 0.2), parameters={(a, b): parameters_ab}, options=ALL_ANGLES_FLOQUET_PHASED_FSIM_CHARACTERIZATION, ) ] )
def test_run_zeta_chi_gamma_calibration_for_moments() -> None: parameters_ab = cirq_google.PhasedFSimCharacterization(zeta=0.5, chi=0.4, gamma=0.3) parameters_bc = cirq_google.PhasedFSimCharacterization(zeta=-0.5, chi=-0.4, gamma=-0.3) parameters_cd = cirq_google.PhasedFSimCharacterization(zeta=0.2, chi=0.3, gamma=0.4) a, b, c, d = cirq.LineQubit.range(4) engine_simulator = cirq_google.PhasedFSimEngineSimulator.create_from_dictionary_sqrt_iswap( parameters={ (a, b): parameters_ab.merge_with(SQRT_ISWAP_PARAMETERS), (b, c): parameters_bc.merge_with(SQRT_ISWAP_PARAMETERS), (c, d): parameters_cd.merge_with(SQRT_ISWAP_PARAMETERS), }) circuit = cirq.Circuit([ [cirq.X(a), cirq.Y(c)], [SQRT_ISWAP_GATE.on(a, b), SQRT_ISWAP_GATE.on(c, d)], [SQRT_ISWAP_GATE.on(b, c)], ]) options = cirq_google.FloquetPhasedFSimCalibrationOptions( characterize_theta=False, characterize_zeta=True, characterize_chi=True, characterize_gamma=True, characterize_phi=False, ) calibrated_circuit, calibrations = workflow.run_zeta_chi_gamma_compensation_for_moments( circuit, engine_simulator, processor_id=None, gate_set=cirq_google.SQRT_ISWAP_GATESET, options=options, ) assert cirq.allclose_up_to_global_phase( engine_simulator.final_state_vector(calibrated_circuit.circuit), cirq.final_state_vector(circuit), ) assert calibrations == [ cirq_google.PhasedFSimCalibrationResult( gate=SQRT_ISWAP_GATE, parameters={ (a, b): parameters_ab, (c, d): parameters_cd }, options=options, ), cirq_google.PhasedFSimCalibrationResult(gate=SQRT_ISWAP_GATE, parameters={ (b, c): parameters_bc }, options=options), ] assert calibrated_circuit.moment_to_calibration == [ None, None, 0, None, None, 1, None ]