From 1393405ed5e8796a37224344c51cdcd4427f9b4b Mon Sep 17 00:00:00 2001
From: Eddy Pedroni <eddy@0xf7.com>
Date: Sun, 5 Jun 2022 11:34:50 +0200
Subject: Minor improvement to automatic vertical scaling

---
 lab_control/test/frequency_response_test.py | 6 +++---
 lab_control/test/mock_lab.py                | 3 +++
 2 files changed, 6 insertions(+), 3 deletions(-)

(limited to 'lab_control/test')

diff --git a/lab_control/test/frequency_response_test.py b/lab_control/test/frequency_response_test.py
index 61fd021..d694887 100644
--- a/lab_control/test/frequency_response_test.py
+++ b/lab_control/test/frequency_response_test.py
@@ -49,19 +49,19 @@ def test_frequencyResponseRamp(uut, mockLab):
 
     expectedData = prepareRampResponse(uut, mockLab)
 
-    assert not uut.measurementDone
-    assert uut.data is None
     assert fg.amplitude is None
     assert fg.function is None
     assert not fg.on
     assert osc.voltsPerDiv is None
+    assert not uut.measurementDone
+    assert uut.data is None
     
     uut.measure(mockLab, mockLab)
 
     assert fg.amplitude == uut.inputAmplitude
     assert fg.function == FunctionGenerator.SINE
     assert fg.on
-    assert osc.voltsPerDiv == uut.inputAmplitude / 6
+    assert osc.voltsPerDiv == uut.inputAmplitude / (mockLab.getDivisionsDisplayed() - 2)
     assert uut.measurementDone
     assert uut.data == expectedData
 
diff --git a/lab_control/test/mock_lab.py b/lab_control/test/mock_lab.py
index 75ba1bf..1016b92 100644
--- a/lab_control/test/mock_lab.py
+++ b/lab_control/test/mock_lab.py
@@ -55,6 +55,9 @@ class MockLab(FunctionGenerator, Oscilloscope):
     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
 
-- 
cgit v1.2.3