1 files changed, 71 insertions, 0 deletions
diff --git a/lab_control/test/mock/mock_lab.py b/lab_control/test/mock/mock_lab.py
new file mode 100644
index 0000000..1016b92
--- /dev/null
+++ b/lab_control/test/mock/mock_lab.py
@@ -0,0 +1,71 @@
+from collections.abc import Callable
+
+from lab_control.function_generator import FunctionGenerator
+from lab_control.oscilloscope import Oscilloscope
+
+class MockLab(FunctionGenerator, Oscilloscope):
+    class FGChannelState:
+        def __init__(self):
+            self.on = False
+            self.frequency = None
+            self.amplitude = None
+            self.function = None
+
+    class OscChannelState:
+        def __init__(self):
+            self.testFunction = None
+            self.connectedChannel = None
+            self.voltsPerDiv = None
+
+    def __init__(self):
+        self.fgChannels = [MockLab.FGChannelState() for i in range(0, 2)]
+        self.oscChannels = [MockLab.OscChannelState() for i in range(0, 4)]
+
+    def setOn(self, channel: int) -> None:
+        self.fgChannels[channel - 1].on = True
+
+    def setOff(self, channel: int) -> None:
+        self.fgChannels[channel - 1].on = False
+
+    def setFrequency(self, channel: int, frequency: float) -> None:
+        self.fgChannels[channel - 1].frequency = frequency
+
+    def setAmplitude(self, channel: int, amplitude: float) -> None:
+        self.fgChannels[channel - 1].amplitude = amplitude
+
+    def setFunction(self, channel: int, function: int) -> None:
+        self.fgChannels[channel - 1].function = function
+
+    def measureAmplitude(self, channel: int) -> float:
+        fgChannel = self.oscChannels[channel - 1].connectedChannel
+        frequency = fgChannel.frequency if fgChannel.on else 0.0
+        amplitude = fgChannel.amplitude if fgChannel.on else 0.0
+
+        return self.oscChannels[channel - 1].testFunction(frequency) * amplitude
+
+    def measurePeakToPeak(self, channel: int) -> float:
+        pass
+
+    def measureRMS(self, channel: int) -> float:
+        pass
+
+    def measureFrequency(self, channel: int) -> float:
+        pass
+
+    def setVoltsPerDivision(self, channel: int, volts: float) -> None:
+        self.oscChannels[channel - 1].voltsPerDiv = volts
+
+    def getDivisionsDisplayed(self) -> int:
+        return 12
+
+    def setTestFunction(self, channel: int, f: Callable[[float], float]) -> None:
+        self.oscChannels[channel - 1].testFunction = f
+
+    def connectChannels(self, fg: int, osc: int) -> None:
+        self.oscChannels[osc - 1].connectedChannel = self.fgChannels[fg - 1]
+
+    def getFunctionGeneratorChannel(self, channel: int):
+        return self.fgChannels[channel - 1]
+
+    def getOscilloscopeChannel(self, channel: int):
+        return self.oscChannels[channel - 1]