From 8bfa50b433a12d926c60e4f9675b166b7b8f3a6e Mon Sep 17 00:00:00 2001
From: Sander de Snoo <59472150+sldesnoo-Delft@users.noreply.github.com>
Date: Wed, 13 Mar 2024 11:20:06 +0100
Subject: [PATCH] add AWG for oscillators in test_configuration.py

---
 pulse_lib/tests/configurations/test_configuration.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/pulse_lib/tests/configurations/test_configuration.py b/pulse_lib/tests/configurations/test_configuration.py
index 5f46121f..3d658d4e 100644
--- a/pulse_lib/tests/configurations/test_configuration.py
+++ b/pulse_lib/tests/configurations/test_configuration.py
@@ -303,6 +303,11 @@ class Context:
                     output = params['output']
                     if not isinstance(output, str):
                         output = tuple(output)
+                    if len(output) == 2:
+                        awg_name = output[0]
+                        if awg_name not in pulse.awg_devices:
+                            pulse.add_awg(station.components[awg_name])
+
                     channel_conf = pulse.digitizer_channels[sensor]
                     channel_conf.iq_out = True
                     dig = pulse.digitizers[channel_conf.module_name]
-- 
GitLab