add pytests

Author: SBlokhuizen

tests/test_waveform.py

@@ -273,3 +273,9 @@ def test_overlap_derivatives():
     expected = [2, 2, -1.5, -1.5, -1.5, -1.5, -1.5]
     values = waveform.get_derivative(np.linspace(0, 3, 7))
     assert np.allclose(values, expected)
+
+
+def test_get_start_end(waveform):
+    """Test if waveform returns correct start and end value."""
+    assert waveform.get_start() == 0
+    assert waveform.get_end() == 14

tests/test_waveform_exporter.py

@@ -0,0 +1,114 @@
+import csv
+from pathlib import Path
+
+import numpy as np
+import pytest
+
+from waveform_editor.waveform import Waveform
+from waveform_editor.waveform_exporter import WaveformExporter
+
+
+@pytest.fixture
+def waveform_list():
+    return [
+        {
+            "user_type": "linear",
+            "user_from": 0,
+            "user_to": 8,
+            "user_duration": 5,
+            "line_number": 1,
+        },
+        {
+            "user_type": "sine-wave",
+            "user_base": 8,
+            "user_amplitude": 2,
+            "user_frequency": 1,
+            "user_duration": 4,
+            "line_number": 2,
+        },
+        {
+            "user_type": "constant",
+            "user_value": 8,
+            "user_duration": 3,
+            "line_number": 3,
+        },
+        {
+            "user_type": "smooth",
+            "user_from": 8,
+            "user_to": 0,
+            "user_duration": 2,
+            "line_number": 4,
+        },
+    ]
+
+
+@pytest.fixture
+def waveform(waveform_list):
+    return Waveform(waveform=waveform_list)
+
+
+@pytest.fixture
+def times():
+    return np.array([0, 1, 2, 3, 4, 5])
+
+
+@pytest.fixture
+def exporter(waveform, times):
+    return WaveformExporter(waveform, times)
+
+
+def test_parse_uri(exporter):
+    """Test if URIs are parsed correctly."""
+    uri = "imas:hdf5?path=./testdb#ec_launchers/beam(1)/power_launched"
+    parsed_uri = exporter.parse_uri(uri)
+    assert parsed_uri == (
+        "imas:hdf5?path=./testdb",
+        "ec_launchers",
+        0,
+        "/beam(1)/power_launched",
+    )
+    uri2 = "imas:hdf5?path=./testdb/test_loc#ec_launchers:1/beam(1)/power_launched"
+    parsed_uri = exporter.parse_uri(uri2)
+    assert parsed_uri == (
+        "imas:hdf5?path=./testdb/test_loc",
+        "ec_launchers",
+        1,
+        "/beam(1)/power_launched",
+    )
+
+    uri3 = "imas:hdf5?path=./test_db#equilibrium/time_slice()/boundary/elongation"
+    parsed_uri = exporter.parse_uri(uri3)
+    assert parsed_uri == (
+        "imas:hdf5?path=./test_db",
+        "equilibrium",
+        0,
+        "/time_slice()/boundary/elongation",
+    )
+
+
+def test_to_csv(exporter, tmp_path):
+    """Test if exported CSV exists and is filled with correct times and values."""
+    file_path = tmp_path / "waveform.csv"
+    exporter.to_csv(file_path)
+
+    with open(file_path, newline="") as csvfile:
+        reader = csv.reader(csvfile)
+        rows = list(reader)
+
+    assert Path(file_path).exists()
+    assert len(rows) == len(exporter.times) + 1  # Header + data rows
+    for i, (time, value) in enumerate(zip(exporter.times, exporter.values), start=1):
+        assert float(rows[i][0]) == time
+        assert float(rows[i][1]) == value
+
+
+def test_to_png(exporter, tmp_path):
+    """Test if exported PNG exists."""
+    file_path = tmp_path / "waveform.png"
+    exporter.to_png(file_path)
+    assert Path(file_path).exists()
+
+
+# TODO: Write tests for exporting to IDS, this requires IMASPy as a dependency
+# def test_to_ids(exporter):
+#     pass