diff options
| author | Eddy Pedroni <eddy@0xf7.com> | 2022-06-03 17:46:06 +0200 |
|---|---|---|
| committer | Eddy Pedroni <eddy@0xf7.com> | 2022-06-03 17:46:06 +0200 |
| commit | 8e9563b7b067f549ea539b56eb14e6624b236acb (patch) | |
| tree | 73eebadbb49475d51ccb206b6a24b4ec59584518 /lab_control/test | |
| parent | 038be6aa9a2882db877954c1a57bb23ea70b5ad5 (diff) | |
Added dump to CSV functionality
Diffstat (limited to 'lab_control/test')
| -rw-r--r-- | lab_control/test/frequency_response_test.py | 19 |
1 files changed, 17 insertions, 2 deletions
diff --git a/lab_control/test/frequency_response_test.py b/lab_control/test/frequency_response_test.py index 014b4f6..ad47d53 100644 --- a/lab_control/test/frequency_response_test.py +++ b/lab_control/test/frequency_response_test.py @@ -37,7 +37,7 @@ def test_frequencyResponseDefaults(uut): assert uut.functionGeneratorChannel == 1 assert uut.oscilloscopeChannel == 1 -def test_frequencyResponseRamp(mockLab, uut): +def test_frequencyResponseRamp(uut, mockLab): uut.minFrequency = 100.0 uut.maxFrequency = 200.0 uut.steps = 11 @@ -49,8 +49,23 @@ def test_frequencyResponseRamp(mockLab, uut): assert not uut.measurementDone assert uut.data == None - response = uut.measure(mockLab, mockLab) + 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]}" |