Waveform-Editor/tests/test_yaml_parser.py at 0b6e027b41103ca344b835fdffbbd53f1898733d · iterorganization/Waveform-Editor · GitHub

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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
from pytest import approx

from waveform_editor.tendencies.constant import ConstantTendency
from waveform_editor.tendencies.linear import LinearTendency
from waveform_editor.tendencies.periodic.sawtooth_wave import SawtoothWaveTendency
from waveform_editor.tendencies.periodic.sine_wave import SineWaveTendency
from waveform_editor.tendencies.periodic.square_wave import SquareWaveTendency
from waveform_editor.tendencies.periodic.triangle_wave import TriangleWaveTendency
from waveform_editor.tendencies.smooth import SmoothTendency
from waveform_editor.yaml_parser import YamlParser


def test_yaml_parser():
    """Test loading a yaml file as a string."""
    # Valid YAML
    with open("tests/tendencies/test_yaml/test.yaml") as file:
        yaml_file = file.read()
    yaml_parser = YamlParser()
    yaml_parser.parse_waveforms(yaml_file)
    assert_tendencies_correct(yaml_parser.waveform.tendencies)
    assert not yaml_parser.waveform.annotations
    assert not yaml_parser.has_yaml_error

    # Invalid configuration
    with open("tests/tendencies/test_yaml/test_invalid_config.yaml") as file:
        yaml_file = file.read()
    yaml_parser = YamlParser()
    yaml_parser.parse_waveforms(yaml_file)
    assert yaml_parser.waveform.annotations
    assert not yaml_parser.has_yaml_error

    # Invalid YAML
    with open("tests/tendencies/test_yaml/test_invalid_yaml.yaml") as file:
        yaml_file = file.read()
    yaml_parser = YamlParser()
    yaml_parser.parse_waveforms(yaml_file)
    assert yaml_parser.waveform.annotations
    assert yaml_parser.has_yaml_error


def assert_tendencies_correct(tendencies):
    """Assert that the tendencies contain the correct parameters."""
    assert isinstance(tendencies[0], LinearTendency)
    assert tendencies[0].start == approx(0)
    assert tendencies[0].end == approx(5)
    assert tendencies[0].duration == approx(5)
    assert tendencies[0].from_ == approx(0)
    assert tendencies[0].to == approx(8)

    assert isinstance(tendencies[1], SineWaveTendency)
    assert tendencies[1].start == approx(5)
    assert tendencies[1].end == approx(9)
    assert tendencies[1].duration == approx(4)
    assert tendencies[1].amplitude == approx(2)
    assert tendencies[1].frequency == approx(1)
    assert tendencies[1].base == approx(8)

    assert isinstance(tendencies[2], SawtoothWaveTendency)
    assert tendencies[2].start == approx(9)
    assert tendencies[2].end == approx(13)
    assert tendencies[2].duration == approx(4)
    assert tendencies[2].amplitude == approx(2)
    assert tendencies[2].frequency == approx(1)
    assert tendencies[2].base == approx(8)

    assert isinstance(tendencies[3], SquareWaveTendency)
    assert tendencies[3].start == approx(13)
    assert tendencies[3].end == approx(17)
    assert tendencies[3].duration == approx(4)
    assert tendencies[3].amplitude == approx(2)
    assert tendencies[3].frequency == approx(1)
    assert tendencies[3].base == approx(8)

    assert isinstance(tendencies[4], TriangleWaveTendency)
    assert tendencies[4].start == approx(17)
    assert tendencies[4].end == approx(21)
    assert tendencies[4].duration == approx(4)
    assert tendencies[4].amplitude == approx(2)
    assert tendencies[4].frequency == approx(1)
    assert tendencies[4].base == approx(8)

    assert isinstance(tendencies[5], ConstantTendency)
    assert tendencies[5].start == approx(21)
    assert tendencies[5].end == approx(24)
    assert tendencies[5].duration == approx(3)
    assert tendencies[5].value == approx(8)

    assert isinstance(tendencies[6], SmoothTendency)
    assert tendencies[6].start == approx(24)
    assert tendencies[6].end == approx(26)
    assert tendencies[6].duration == approx(2)
    assert tendencies[6].from_ == approx(8)
    assert tendencies[6].to == approx(0)


def test_scientific_notation():
    """Test if scientific notation is parsed correctly."""
    waveforms = {
        "waveform:\n- {type: linear, to: 1.5e5}": 1.5e5,
        "waveform:\n- {type: linear, to: 1.5e+5}": 1.5e5,
        "waveform:\n- {type: linear, to: 1.5E+5}": 1.5e5,
        "waveform:\n- {type: linear, to: 1.5e-5}": 1.5e-5,
        "waveform:\n- {type: linear, to: 1.5E-5}": 1.5e-5,
        "waveform:\n- {type: linear, to: 1e5}": 1e5,
        "waveform:\n- {type: linear, to: 1e+5}": 1e5,
        "waveform:\n- {type: linear, to: 1E+5}": 1e5,
        "waveform:\n- {type: linear, to: 1e-5}": 1e-5,
        "waveform:\n- {type: linear, to: 1E-5}": 1e-5,
        "waveform:\n- {type: linear, to: 0e5}": 0.0,
        "waveform:\n- {type: linear, to: 0E0}": 0.0,
        "waveform:\n- {type: linear, to: -1.5e5}": -1.5e5,
        "waveform:\n- {type: linear, to: -1E+5}": -1e5,
        "waveform:\n- {type: linear, to: -1.5e-5}": -1.5e-5,
    }

    yaml_parser = YamlParser()

    for waveform, expected_value in waveforms.items():
        yaml_parser.parse_waveforms(waveform)
        assert yaml_parser.waveform.tendencies[0].to == expected_value