From 1aec3bacaab2991121e0c00d147fd418cdbfbc17 Mon Sep 17 00:00:00 2001
From: Eddy Pedroni <eddy@0xf7.com>
Date: Fri, 3 Jun 2022 17:47:50 +0200
Subject: Improved CSV dump
---
lab_control/frequency_response.py | 3 +++
lab_control/test/frequency_response_test.py | 6 +++++-
2 files changed, 8 insertions(+), 1 deletion(-)
(limited to 'lab_control')
diff --git a/lab_control/frequency_response.py b/lab_control/frequency_response.py
index af78840..6508d80 100644
--- a/lab_control/frequency_response.py
+++ b/lab_control/frequency_response.py
@@ -26,6 +26,9 @@ class FrequencyResponseMeasurement(Measurement):
self.measurementDone = True
def dumpToCSV(self, path: Path) -> None:
+ if self.data is None:
+ return
+
with open(path, "w") as f:
f.write("Frequency (Hz),Amplitude (V)")
diff --git a/lab_control/test/frequency_response_test.py b/lab_control/test/frequency_response_test.py
index ad47d53..2997413 100644
--- a/lab_control/test/frequency_response_test.py
+++ b/lab_control/test/frequency_response_test.py
@@ -58,11 +58,15 @@ def test_dumpToCSV(uut, mockLab, tmp_path):
uut.minFrequency = 10.0
uut.maxFrequency = 30.0
uut.steps = 5
+ csv = tmp_path / "data.csv"
+
+ assert not csv.exists()
+ uut.dumpToCSV(csv)
+ assert not csv.exists()
expectedData = prepareRampResponse(uut, mockLab)
uut.measure(mockLab, mockLab)
- csv = tmp_path / "data.csv"
uut.dumpToCSV(csv)
savedData = csv.read_text().split("\n")
--
cgit v1.2.3