Skip to content

Commit f46782c

Browse files
committed
Fix tests
1 parent 5c2450e commit f46782c

4 files changed

Lines changed: 92 additions & 13 deletions

File tree

tests/test_preisstaffel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ class TestPreisstaffel:
1414
[
1515
pytest.param(
1616
Preisstaffel(
17-
einheitspreis=Decimal(40.0),
17+
preis=Decimal(40.0),
1818
staffelgrenze_von=Decimal(12.5),
1919
staffelgrenze_bis=Decimal(25.0),
2020
sigmoidparameter=Sigmoidparameter(),

tests/test_region.py

Lines changed: 31 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import pytest
22

3-
from bo4e import Region, Regionskriterium
3+
from bo4e import Operator, Region, Regionskriterium, Regionsoperation
44
from tests.serialization_helper import assert_serialization_roundtrip
55

66

@@ -11,12 +11,38 @@ class TestRegion:
1111
pytest.param(
1212
Region(
1313
bezeichnung="Bikini Bottom",
14-
positiv_liste=[Regionskriterium()],
15-
negativ_liste=[Regionskriterium()],
14+
beschreibung="Heimatort der Krabbencrew",
15+
regionsoperationen=[
16+
Regionsoperation(
17+
regionsoperator=Operator.ADDITION,
18+
prioritaet=0,
19+
bezeichnung="Bikini Bottom",
20+
regionskriterium=Regionskriterium.ORT,
21+
),
22+
],
1623
),
17-
id="max attributes",
24+
id="single regionsoperation",
25+
),
26+
pytest.param(
27+
Region(
28+
bezeichnung="NRW ohne Düsseldorf",
29+
regionsoperationen=[
30+
Regionsoperation(
31+
regionsoperator=Operator.ADDITION,
32+
prioritaet=0,
33+
bezeichnung="NRW",
34+
regionskriterium=Regionskriterium.BUNDESLAND_NAME,
35+
),
36+
Regionsoperation(
37+
regionsoperator=Operator.SUBTRAKTION,
38+
prioritaet=1,
39+
bezeichnung="Düsseldorf",
40+
regionskriterium=Regionskriterium.ORT,
41+
),
42+
],
43+
),
44+
id="combined regionsoperationen",
1845
),
19-
pytest.param(Region(bezeichnung="Bikini Bottom", positiv_liste=[Regionskriterium()]), id="min attributes"),
2046
],
2147
)
2248
def test_serialization_roundtrip(self, region: Region) -> None:

tests/test_tarif.py

Lines changed: 59 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,22 +1,34 @@
11
from datetime import datetime, timezone
2+
from decimal import Decimal
23

34
import pytest
45

56
from bo4e import (
6-
AufAbschlagRegional,
7+
EinheitsPreisposition,
78
Energiemix,
89
Kundentyp,
910
Marktteilnehmer,
11+
Mengeneinheit,
12+
Operator,
13+
Preis,
1014
Preisgarantie,
15+
Preisreferenz,
16+
Region,
17+
Regionskriterium,
18+
Regionsoperation,
19+
Regionspreis,
20+
Regionszeitscheibe,
1121
Registeranzahl,
22+
RelativePreisposition,
1223
Sparte,
1324
Tarif,
1425
Tarifberechnungsparameter,
1526
Tarifeinschraenkung,
1627
Tarifmerkmal,
17-
TarifpreispositionProOrt,
28+
Tarifpreiszeitscheibe,
1829
Tariftyp,
1930
Vertragskonditionen,
31+
Waehrungseinheit,
2032
Zeitraum,
2133
)
2234
from tests.serialization_helper import assert_serialization_roundtrip
@@ -28,10 +40,7 @@ class TestTarif:
2840
[
2941
pytest.param(
3042
Tarif(
31-
preisstand=datetime(2022, 2, 1, 0, 0, 0, tzinfo=timezone.utc),
3243
berechnungsparameter=Tarifberechnungsparameter(),
33-
tarif_auf_abschlaege=[AufAbschlagRegional()],
34-
tarifpreise=[TarifpreispositionProOrt()],
3544
preisgarantie=Preisgarantie(),
3645
tarifeinschraenkung=Tarifeinschraenkung(),
3746
# below are the attributes of tarifinfo
@@ -46,8 +55,52 @@ class TestTarif:
4655
bemerkung="super billig aber auch super dreckig",
4756
vertragskonditionen=Vertragskonditionen(),
4857
zeitliche_gueltigkeit=Zeitraum(),
49-
energiemix=Energiemix(),
58+
energiemix=[Energiemix()],
5059
anbieter=Marktteilnehmer(),
60+
regionspreise=[
61+
Regionspreis(
62+
regionszeitscheiben=[
63+
Regionszeitscheibe(
64+
zeitscheibengueltigkeit=Zeitraum(startdatum=datetime(2020, 1, 1)),
65+
region=Region(
66+
bezeichnung="Deutschland",
67+
regionsoperationen=[
68+
Regionsoperation(
69+
regionsoperator=Operator.ADDITION,
70+
prioritaet=0,
71+
bezeichnung="Deutschland",
72+
regionskriterium=Regionskriterium.BUNDESWEIT,
73+
)
74+
],
75+
),
76+
),
77+
],
78+
tarifpreiszeitscheiben=[
79+
Tarifpreiszeitscheibe(
80+
zeitscheibengueltigkeit=Zeitraum(startdatum=datetime(2020, 1, 1)),
81+
einheits_preispositionen=[
82+
EinheitsPreisposition(
83+
id="12345",
84+
bezeichnung="Arbeitspreis",
85+
preisreferenz=Preisreferenz.ENERGIEMENGE,
86+
preis=Preis(
87+
wert=Decimal("30"),
88+
einheit=Waehrungseinheit.CT,
89+
bezugswert=Mengeneinheit.KWH,
90+
),
91+
)
92+
],
93+
relative_preispositionen=[
94+
RelativePreisposition(
95+
bezeichnung="5% Rabatt auf Arbeitspreis",
96+
id_referenz="12345",
97+
wert=Decimal("0.95"),
98+
)
99+
],
100+
)
101+
],
102+
)
103+
],
51104
),
52105
id="all attributes",
53106
),

tests/test_tarifpreisposition.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ class TestTarifpreisposition:
1717
bezugseinheit=Mengeneinheit.KWH,
1818
preisstaffeln=[
1919
Preisstaffel(
20-
einheitspreis=Decimal(40.0),
20+
preis=Decimal(40.0),
2121
staffelgrenze_von=Decimal(12.5),
2222
staffelgrenze_bis=Decimal(25.0),
2323
),

0 commit comments

Comments
 (0)