1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
import pytest
from lab_control.test.mock_lab import MockLab
from lab_control.frequency_response import FrequencyResponseMeasurement
@pytest.fixture
def mockLab():
return MockLab()
@pytest.fixture
def uut(mockLab):
return FrequencyResponseMeasurement()
def test_frequencyResponseDefaults(uut):
assert uut.minFrequency == 20e0
assert uut.maxFrequency == 16e3
assert uut.steps == 50
assert uut.functionGeneratorChannel == 1
assert uut.oscilloscopeChannel == 1
def test_frequencyResponseRamp(mockLab, uut):
uut.minFrequency = 100.0
uut.maxFrequency = 200.0
uut.steps = 11
uut.functionGeneratorChannel = 1
uut.oscilloscopeChannel = 1
# Expect a ramp response from 0.5 to 1.5 * input amplitude
inputAmplitude = 1.0
minScale = 0.5
maxScale = 1.5
def testFunction(f: float) -> float:
assert f >= uut.minFrequency and f <= uut.maxFrequency
frequencyPu = (f - uut.minFrequency) / (uut.maxFrequency - uut.minFrequency)
return minScale + maxScale * frequencyPu
mockLab.connectChannels(uut.functionGeneratorChannel, uut.oscilloscopeChannel)
mockLab.setTestFunction(uut.oscilloscopeChannel, testFunction)
mockLab.setAmplitude(uut.functionGeneratorChannel, inputAmplitude)
mockLab.setOn(uut.functionGeneratorChannel)
expectedData = [(f, testFunction(f) * inputAmplitude) for f in [100.0, 110.0, 120.0, 130.0, 140.0, 150.0, 160.0, 170.0, 180.0, 190.0, 200.0]]
assert not uut.measurementDone
assert uut.data == None
response = uut.measure(mockLab, mockLab)
assert uut.measurementDone
assert uut.data == expectedData
|
