Added tests

pulse_lib/tests/baseband/test_sweep.py

+
+from pulse_lib.tests.configurations.test_configuration import context
+import pulse_lib.segments.utility.looping as lp
+
+import matplotlib.pyplot as pt
+import numpy as np
+
+#%%
+def test1():
+    pulse = context.init_pulselib(n_gates=1)
+    context.station.AWG1.set_digital_filter_mode(1)
+
+    segment = pulse.mk_segment(hres=True)
+
+    amplitude = lp.linspace(100.0, 200.0, 21,
+                            name='amplitude', unit='mV')
+    segment.P1.add_block(5, 15, amplitude)
+
+
+    s.P1.add_ramp_ss(15, 18, 80, 0)
+    s.reset_time()
+    s.P1.add_sin(10, 40, 50, 350e6)
+
+    s.wait(20)
+    sequence = pulse.mk_sequence([s])
+    sequence.n_rep = None
+    context.add_hw_schedule(sequence)
+
+    context.plot_awgs(sequence, analogue_out=True, ylim=(-0.1,0.100), xlim=(0, 80))
+
+#%%
+if __name__ == '__main__':
+    ds1 = test1()

pulse_lib/tests/test_iq/test_iq_corrections.py

+
+from pulse_lib.tests.configurations.test_configuration import context
+
+from numpy import pi
+
+#%%
+def test1():
+    pulse = context.init_pulselib(n_qubits=1)
+    pulse.set_qubit_correction_phase('q1', 0.1*pi)
+
+    s = pulse.mk_segment()
+
+    s.q1.add_MW_pulse(0, 20, 100, 2.450e9)
+
+    context.plot_segments([s])
+
+    sequence = pulse.mk_sequence([s])
+    sequence.n_rep = 1
+    context.add_hw_schedule(sequence)
+    context.plot_awgs(sequence)
+
+    return None
+
+def test2():
+    pulse = context.init_pulselib(n_qubits=1)
+    pulse.set_qubit_correction_gain('q1', 1.0, 1.1)
+
+    s = pulse.mk_segment()
+
+    s.q1.add_MW_pulse(0, 20, 100, 2.450e9)
+
+    context.plot_segments([s])
+
+    sequence = pulse.mk_sequence([s])
+    sequence.n_rep = 1
+    context.add_hw_schedule(sequence)
+    context.plot_awgs(sequence)
+
+    return None
+
+if __name__ == '__main__':
+    ds1 = test1()
+    ds2 = test2()

pulse_lib/tests/virtual_gates/test_virtual_gates.py

+
+from pulse_lib.tests.configurations.test_configuration import context
+import pulse_lib.segments.utility.looping as lp
+
+#%%
+def test1():
+    pulse = context.init_pulselib(n_gates=2, virtual_gates=True)
+
+    pulse.add_channel_attenuation('P1', 0.1)
+    pulse.add_channel_attenuation('P2', 0.1)
+
+    s = pulse.mk_segment()
+
+    s.vP1.add_block(0,100, -10)
+    s.vP1.add_ramp_ss(4, 8, 0, 80)
+    s.vP1.add_block(8, 10, 80)
+    s.vP1.add_ramp_ss(10, 14, 80, 0)
+    s.vP2.wait(20)
+    s.vP2.reset_time()
+    s.vP2.add_ramp_ss(4, 8, 0, 80)
+    s.vP2.add_block(8, 10, 80)
+    s.vP2.add_ramp_ss(10, 14, 80, 0)
+
+    s.vP1.add_ramp_ss(60, 90, 20, 40)
+    s.vP1.add_ramp_ss(70, 80, 0, -40)
+    s.reset_time()
+#    s.wait(100000)
+
+    sequence = pulse.mk_sequence([s])
+#    sequence.n_rep = 10000
+    sequence.n_rep = None
+    context.add_hw_schedule(sequence)
+
+    context.plot_awgs(sequence, ylim=(-0.2,1.100), xlim=(0, 100))
+
+#    return context.run('hres1', sequence)
+
+
+#%%
+if __name__ == '__main__':
+    ds1 = test1()