From b543dda207da16aab3dfd7a6f322fb184c214087 Mon Sep 17 00:00:00 2001
From: sldesnoo-Delft <s.l.desnoo@tudelft.nl>
Date: Mon, 8 May 2023 16:17:21 +0200
Subject: [PATCH] Extended MockM3102A

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

diff --git a/pulse_lib/tests/mock_m3102a.py b/pulse_lib/tests/mock_m3102a.py
index bbd7bebb..721d15ac 100644
--- a/pulse_lib/tests/mock_m3102a.py
+++ b/pulse_lib/tests/mock_m3102a.py
@@ -30,6 +30,10 @@ class MockM3102A(Instrument):
     def set_active_channels(self, channel_list):
         self.measure._active_channels = set(channel_list)
 
+    def set_data_handling_mode(self, mode):
+        for ch in self.measure._active_channels:
+            self.measure._ch_properties[ch].data_mode = mode
+
     @property
     def active_channels(self):
         return self.measure._active_channels
@@ -70,6 +74,7 @@ class ChannelProperties:
     n_cycles: int = 1
     t_measure: int = 10
     samples_per_cycle: int = 1
+    data_mode: int = 0
 
 class ChannelData:
     def __init__(self):
-- 
GitLab