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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
import pytest
from lab_control.test.mock_lab import MockLab
from lab_control.frequency_response import FrequencyResponseMeasurement
from lab_control.measurement import getLinearRange
@pytest.fixture
def mockLab():
return MockLab()
@pytest.fixture
def uut(mockLab):
return FrequencyResponseMeasurement()
def prepareRampResponse(uut, mockLab):
# Expect a ramp response from 0.5 to 1.5 * input amplitude
inputAmplitude = 2.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)
return [(f, testFunction(f) * inputAmplitude) for f in getLinearRange(uut.minFrequency, uut.maxFrequency, uut.steps)]
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(uut, mockLab):
uut.minFrequency = 100.0
uut.maxFrequency = 200.0
uut.steps = 11
uut.functionGeneratorChannel = 1
uut.oscilloscopeChannel = 1
expectedData = prepareRampResponse(uut, mockLab)
assert not uut.measurementDone
assert uut.data == None
uut.measure(mockLab, mockLab)
assert uut.measurementDone
assert uut.data == expectedData
def test_dumpToCSV(uut, mockLab, tmp_path):
uut.minFrequency = 10.0
uut.maxFrequency = 30.0
uut.steps = 5
expectedData = prepareRampResponse(uut, mockLab)
uut.measure(mockLab, mockLab)
csv = tmp_path / "data.csv"
uut.dumpToCSV(csv)
savedData = csv.read_text().split("\n")
assert savedData[0] == "Frequency (Hz),Amplitude (V)"
for i, l in enumerate(savedData[1:]):
assert l == f"{expectedData[i][0]},{expectedData[i][1]}"
|
