Skip to content

Commit ab8e1fe

Browse files
committed
fix failing pytests due to new DD version
1 parent 445c053 commit ab8e1fe

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

tests/test_configuration.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from waveform_editor.tendencies.linear import LinearTendency
88
from waveform_editor.tendencies.periodic.sine_wave import SineWaveTendency
99
from waveform_editor.tendencies.smooth import SmoothTendency
10+
from waveform_editor.util import LATEST_DD_VERSION
1011
from waveform_editor.waveform import Waveform
1112
from waveform_editor.yaml_parser import YamlParser
1213

@@ -340,7 +341,7 @@ def test_load_yaml_globals():
340341
ec_launchers/beam(1)/phase/angle: 1e-3
341342
"""
342343
config.load_yaml(yaml_str)
343-
assert config.globals.dd_version == "4.0.0"
344+
assert config.globals.dd_version == LATEST_DD_VERSION
344345
assert not config.globals.machine_description
345346

346347

tests/test_yaml_parser.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
from waveform_editor.tendencies.periodic.square_wave import SquareWaveTendency
1111
from waveform_editor.tendencies.periodic.triangle_wave import TriangleWaveTendency
1212
from waveform_editor.tendencies.smooth import SmoothTendency
13+
from waveform_editor.util import LATEST_DD_VERSION
1314
from waveform_editor.waveform import Waveform
1415
from waveform_editor.yaml_parser import YamlParser
1516

@@ -216,7 +217,7 @@ def test_load_yaml_globals_missing_dd_version(yaml_parser, config):
216217
yaml_parser.load_yaml(yaml_str)
217218
assert not config.groups
218219
assert not config.waveform_map
219-
assert config.globals.dd_version == "4.0.0"
220+
assert config.globals.dd_version == LATEST_DD_VERSION
220221
assert (
221222
config.globals.machine_description["ec_launchers"] == "imas:hdf5?path=test_md"
222223
)

0 commit comments

Comments
 (0)