From 6b6919227cf3159fbb19da6880b8329e9adfe24b Mon Sep 17 00:00:00 2001
From: Sander de Snoo <59472150+sldesnoo-Delft@users.noreply.github.com>
Date: Fri, 13 Oct 2023 15:45:46 +0200
Subject: [PATCH] Fixed test
---
pulse_lib/tests/baseband/test_custom.py | 24 ++++++++++++++----------
1 file changed, 14 insertions(+), 10 deletions(-)
diff --git a/pulse_lib/tests/baseband/test_custom.py b/pulse_lib/tests/baseband/test_custom.py
index 491ab0cb..5e0032da 100644
--- a/pulse_lib/tests/baseband/test_custom.py
+++ b/pulse_lib/tests/baseband/test_custom.py
@@ -1,7 +1,7 @@
from pulse_lib.tests.configurations.test_configuration import context
-#%%
+# %%
from scipy import signal
@@ -21,7 +21,8 @@ def tukey_pulse(duration, sample_rate, amplitude, alpha):
n_points = int(round(duration / sample_rate * 1e9))
return signal.windows.tukey(n_points, alpha) * amplitude
-#%%
+# %%
+
def test1():
pulse = context.init_pulselib(n_gates=1, n_markers=1)
@@ -40,9 +41,10 @@ def test1():
sequence = pulse.mk_sequence(segments)
sequence.n_rep = 2
-
+ context.add_hw_schedule(sequence)
context.plot_awgs(sequence, ylim=(-0.2, 0.2))
+
def test2():
pulse = context.init_pulselib(n_gates=1, n_markers=1)
@@ -61,7 +63,7 @@ def test2():
sequence = pulse.mk_sequence(segments)
sequence.n_rep = 2
-
+ context.add_hw_schedule(sequence)
context.plot_awgs(sequence, ylim=(-0.2, 0.2))
@@ -83,7 +85,7 @@ def test3():
sequence = pulse.mk_sequence(segments)
sequence.n_rep = 2
-
+ context.add_hw_schedule(sequence)
context.plot_awgs(sequence, ylim=(-0.2, 0.2))
@@ -106,9 +108,10 @@ def test4():
sequence = pulse.mk_sequence(segments)
sequence.n_rep = 2
-
+ context.add_hw_schedule(sequence)
context.plot_awgs(sequence, ylim=(-0.2, 0.2))
+
def test5():
pulse = context.init_pulselib(n_gates=1, n_markers=1)
@@ -129,10 +132,11 @@ def test5():
sequence = pulse.mk_sequence(segments)
sequence.n_rep = 2
-
+ context.add_hw_schedule(sequence)
context.plot_awgs(sequence, ylim=(-0.2, 0.2))
-#%%
+
+# %%
if __name__ == '__main__':
ds1 = test1()
ds2 = test2()
@@ -140,8 +144,8 @@ if __name__ == '__main__':
ds4 = test4()
ds5 = test5()
-#%%
+# %%
if False:
from pulse_lib.tests.utils.last_upload import get_last_upload
- lu = get_last_upload(context.pulse)
\ No newline at end of file
+ lu = get_last_upload(context.pulse)
--
GitLab