From d37252c9b6044ce540db54e8b480a85dfc1ec25d Mon Sep 17 00:00:00 2001 From: Eddy Pedroni Date: Sun, 5 Jun 2022 16:49:33 +0200 Subject: Moved mocks to separate module --- lab_control/test/mock_sds1000xe_device.py | 47 ------------------------------- 1 file changed, 47 deletions(-) delete mode 100644 lab_control/test/mock_sds1000xe_device.py (limited to 'lab_control/test/mock_sds1000xe_device.py') diff --git a/lab_control/test/mock_sds1000xe_device.py b/lab_control/test/mock_sds1000xe_device.py deleted file mode 100644 index 04ec07a..0000000 --- a/lab_control/test/mock_sds1000xe_device.py +++ /dev/null @@ -1,47 +0,0 @@ -import re - -class MockSDS1000XEDevice: - def __init__(self): - # Mock internal values - self._channels = [{"AMPL" : None, "VDIV" : None} for i in range(0, 4)] - - def _handleRequest(self, request: str) -> str: - m = re.search(r"C(?P\d):(?P\w+)\??\s(?P.+)", request.strip()) - if not m: - return None - - channelIndex = int(m.group("channel")) - 1 - opcode = m.group("opcode") - - if opcode == "PAVA": - arg = m.group("arg") - value = self._channels[channelIndex].get(arg) - unit = "Hz" if arg == "FREQ" else "V" - - if value is None: - return None - else: - response = f"C{m.group('channel')}:PAVA {arg},{value:.6E}{unit}" - return response - elif opcode == "VDIV": - arg = float(m.group("arg").rstrip("V")) - self._channels[channelIndex]["VDIV"] = arg - return None - - return None - - def setAmplitude(self, channel: int, value: float) -> None: - self._channels[channel - 1]["AMPL"] = value - - def setPeakToPeak(self, channel: int, value: float) -> None: - self._channels[channel - 1]["PKPK"] = value - - def setRMS(self, channel: int, value: float) -> None: - self._channels[channel - 1]["RMS"] = value - - def setFrequency(self, channel: int, value: float) -> None: - self._channels[channel - 1]["FREQ"] = value - - def getVoltsPerDivision(self, channel: int) -> float: - return self._channels[channel - 1]["VDIV"] - -- cgit v1.2.3