From 7a8804f4c0d0e5ab19c4fa9816f795c51922192e Mon Sep 17 00:00:00 2001
From: Sander de Snoo <59472150+sldesnoo-Delft@users.noreply.github.com>
Date: Wed, 13 Mar 2024 11:45:34 +0100
Subject: [PATCH] Add set_channel_acquisition_mode to mock

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

diff --git a/pulse_lib/tests/mock_m3102a.py b/pulse_lib/tests/mock_m3102a.py
index 7218c9d1..486c1361 100644
--- a/pulse_lib/tests/mock_m3102a.py
+++ b/pulse_lib/tests/mock_m3102a.py
@@ -28,6 +28,9 @@ class MockM3102A(Instrument):
         for prop in self.measure._ch_properties.values():
             prop.acquisition_mode = mode
 
+    def set_channel_acquisition_mode(self, ch_num, mode):
+        self.measure._ch_properties[ch_num].acquisition_mode = mode
+
     def get_channel_acquisition_mode(self, ch_num):
         return self.measure._ch_properties[ch_num].acquisition_mode
 
-- 
GitLab