def TestControlAmpNoise(self): """ Test for the control amplitude noise """ tlist = np.array([1, 2, 3, 4, 5, 6]) coeff = np.array([1, 1, 1, 1, 1, 1]) # use proc_qobjevo pulses = [Pulse(sigmaz(), 0, tlist, coeff)] connoise = ControlAmpNoise(coeff=coeff, tlist=tlist) noisy_pulses = connoise.get_noisy_dynamics(pulses=pulses) assert_allclose(noisy_pulses[0].coherent_noise[0].qobj, sigmaz()) assert_allclose(noisy_pulses[0].coherent_noise[0].coeff, coeff)
def TestGetObjevo(self): tlist = np.array([1, 2, 3, 4, 5, 6], dtype=float) coeff = np.array([1, 1, 1, 1, 1, 1], dtype=float) processor = Processor(N=1) processor.add_control(sigmaz()) processor.pulses[0].tlist = tlist processor.pulses[0].coeff = coeff # without noise unitary_qobjevo, _ = processor.get_qobjevo( args={"test": True}, noisy=False) assert_allclose(unitary_qobjevo.ops[0].qobj, sigmaz()) assert_allclose(unitary_qobjevo.tlist, tlist) assert_allclose(unitary_qobjevo.ops[0].coeff, coeff[0]) assert_(unitary_qobjevo.args["test"], msg="Arguments not correctly passed on") # with decoherence noise dec_noise = DecoherenceNoise( c_ops=sigmax(), coeff=coeff, tlist=tlist) processor.add_noise(dec_noise) assert_equal(unitary_qobjevo.to_list(), processor.get_qobjevo(noisy=False)[0].to_list()) noisy_qobjevo, c_ops = processor.get_qobjevo( args={"test": True}, noisy=True) assert_(noisy_qobjevo.args["_step_func_coeff"], msg="Spline type not correctly passed on") assert_(noisy_qobjevo.args["test"], msg="Arguments not correctly passed on") assert_(sigmaz() in [pair[0] for pair in noisy_qobjevo.to_list()]) assert_equal(c_ops[0].ops[0].qobj, sigmax()) assert_equal(c_ops[0].tlist, tlist) # with amplitude noise processor = Processor(N=1, spline_kind="cubic") processor.add_control(sigmaz()) tlist = np.linspace(1, 6, int(5/0.2)) coeff = np.random.rand(len(tlist)) processor.pulses[0].tlist = tlist processor.pulses[0].coeff = coeff amp_noise = ControlAmpNoise(coeff=coeff, tlist=tlist) processor.add_noise(amp_noise) noisy_qobjevo, c_ops = processor.get_qobjevo( args={"test": True}, noisy=True) assert_(not noisy_qobjevo.args["_step_func_coeff"], msg="Spline type not correctly passed on") assert_(noisy_qobjevo.args["test"], msg="Arguments not correctly passed on") assert_equal(len(noisy_qobjevo.ops), 2) assert_equal(sigmaz(), noisy_qobjevo.ops[0].qobj) assert_allclose(coeff, noisy_qobjevo.ops[0].coeff, rtol=1.e-10)
def TestSpline(self): """ Test if the spline kind is correctly transfered into the arguments in QobjEvo """ tlist = np.array([1, 2, 3, 4, 5, 6], dtype=float) coeff = np.array([1, 1, 1, 1, 1, 1], dtype=float) processor = Processor(N=1, spline_kind="step_func") processor.add_control(sigmaz()) processor.pulses[0].tlist = tlist processor.pulses[0].coeff = coeff ideal_qobjevo, _ = processor.get_qobjevo(noisy=False) assert_(ideal_qobjevo.args["_step_func_coeff"]) noisy_qobjevo, c_ops = processor.get_qobjevo(noisy=True) assert_(noisy_qobjevo.args["_step_func_coeff"]) processor.T1 = 100. processor.add_noise(ControlAmpNoise(coeff=coeff, tlist=tlist)) noisy_qobjevo, c_ops = processor.get_qobjevo(noisy=True) assert_(noisy_qobjevo.args["_step_func_coeff"]) tlist = np.array([1, 2, 3, 4, 5, 6], dtype=float) coeff = np.array([1, 1, 1, 1, 1, 1], dtype=float) processor = Processor(N=1, spline_kind="cubic") processor.add_control(sigmaz()) processor.pulses[0].tlist = tlist processor.pulses[0].coeff = coeff ideal_qobjevo, _ = processor.get_qobjevo(noisy=False) assert_(not ideal_qobjevo.args["_step_func_coeff"]) noisy_qobjevo, c_ops = processor.get_qobjevo(noisy=True) assert_(not noisy_qobjevo.args["_step_func_coeff"]) processor.T1 = 100. processor.add_noise(ControlAmpNoise(coeff=coeff, tlist=tlist)) noisy_qobjevo, c_ops = processor.get_qobjevo(noisy=True) assert_(not noisy_qobjevo.args["_step_func_coeff"])