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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
|
"""Test the models."""
from datetime import UTC, date, datetime
import pytest
from aiohttp import ClientSession
from aresponses import ResponsesMockServer
from easyenergy import EasyEnergy, EasyEnergyNoDataError, Electricity, Gas, VatOption
from . import load_fixtures
@pytest.mark.freeze_time("2022-12-29 15:00:00+01:00")
async def test_electricity_model_usage(aresponses: ResponsesMockServer) -> None:
"""Test the electricity model for usage at 15:00:00 CET."""
aresponses.add(
"mijn.easyenergy.com",
"/nl/api/tariff/getapxtariffs",
"GET",
aresponses.Response(
status=200,
headers={"Content-Type": "application/json"},
text=load_fixtures("energy.json"),
),
)
async with ClientSession() as session:
today = date(2022, 12, 29)
client = EasyEnergy(session=session)
energy: Electricity = await client.energy_prices(
start_date=today,
end_date=today,
vat=VatOption.INCLUDE,
)
assert energy is not None
assert isinstance(energy, Electricity)
assert energy.extreme_usage_prices[1] == 0.13345
assert energy.extreme_usage_prices[0] == -0.00277
assert energy.average_usage_price == 0.06941
assert energy.current_usage_price == 0.1199
# The next hour price
next_hour = datetime(2022, 12, 29, 15, 0, tzinfo=UTC)
assert energy.price_at_time(next_hour) == 0.11979
assert energy.lowest_usage_price_time == datetime.strptime(
"2022-12-29 02:00",
"%Y-%m-%d %H:%M",
).replace(tzinfo=UTC)
assert energy.highest_usage_price_time == datetime.strptime(
"2022-12-29 17:00",
"%Y-%m-%d %H:%M",
).replace(tzinfo=UTC)
assert energy.pct_of_max_usage == 89.85
assert isinstance(energy.timestamp_usage_prices, list)
assert energy.hours_priced_equal_or_lower_usage == 21
assert energy.hours_priced_equal_or_higher_return == 5
@pytest.mark.freeze_time("2022-12-29 15:00:00+01:00")
async def test_electricity_model_return(aresponses: ResponsesMockServer) -> None:
"""Test the electricity model for return at 15:00:00 CET."""
aresponses.add(
"mijn.easyenergy.com",
"/nl/api/tariff/getapxtariffs",
"GET",
aresponses.Response(
status=200,
headers={"Content-Type": "application/json"},
text=load_fixtures("energy.json"),
),
)
async with ClientSession() as session:
today = date(2022, 12, 29)
client = EasyEnergy(session=session)
energy: Electricity = await client.energy_prices(
start_date=today,
end_date=today,
vat=VatOption.INCLUDE,
)
assert energy is not None
assert isinstance(energy, Electricity)
assert energy.extreme_return_prices[1] == 0.12243
assert energy.extreme_return_prices[0] == -0.00254
assert energy.average_return_price == 0.06368
assert energy.current_return_price == 0.11
# The next hour price
next_hour = datetime(2022, 12, 29, 15, 0, tzinfo=UTC)
assert energy.price_at_time(next_hour, data_type="return") == 0.1099
assert energy.lowest_return_price_time == datetime.strptime(
"2022-12-29 02:00",
"%Y-%m-%d %H:%M",
).replace(tzinfo=UTC)
assert energy.highest_return_price_time == datetime.strptime(
"2022-12-29 17:00",
"%Y-%m-%d %H:%M",
).replace(tzinfo=UTC)
assert energy.pct_of_max_return == 89.85
assert isinstance(energy.timestamp_return_prices, list)
@pytest.mark.freeze_time("2022-12-29 00:30:00+02:00")
async def test_electricity_midnight(aresponses: ResponsesMockServer) -> None:
"""Test the electricity model between 00:00 and 01:00 with in CEST."""
aresponses.add(
"mijn.easyenergy.com",
"/nl/api/tariff/getapxtariffs",
"GET",
aresponses.Response(
status=200,
headers={"Content-Type": "application/json"},
text=load_fixtures("energy.json"),
),
)
async with ClientSession() as session:
today = date(2022, 12, 29)
client = EasyEnergy(session=session)
energy: Electricity = await client.energy_prices(
start_date=today,
end_date=today,
)
assert energy is not None
assert energy.current_usage_price == 0.02341
assert energy.current_return_price == 0.02148
async def test_electricity_none_data(aresponses: ResponsesMockServer) -> None:
"""Test when there is no data for the current datetime."""
aresponses.add(
"mijn.easyenergy.com",
"/nl/api/tariff/getapxtariffs",
"GET",
aresponses.Response(
status=200,
headers={"Content-Type": "application/json"},
text=load_fixtures("energy.json"),
),
)
async with ClientSession() as session:
today = date(2022, 12, 29)
client = EasyEnergy(session=session)
energy: Electricity = await client.energy_prices(
start_date=today,
end_date=today,
)
assert energy is not None
assert isinstance(energy, Electricity)
assert energy.current_return_price is None
assert energy.average_return_price == 0.06368
async def test_no_electricity_data(aresponses: ResponsesMockServer) -> None:
"""Test when there is no electricity data."""
aresponses.add(
"mijn.easyenergy.com",
"/nl/api/tariff/getapxtariffs",
"GET",
aresponses.Response(
status=200,
headers={"Content-Type": "application/json"},
text="[]",
),
)
async with ClientSession() as session:
today = date(2022, 12, 16)
client = EasyEnergy(session=session)
with pytest.raises(EasyEnergyNoDataError):
await client.energy_prices(start_date=today, end_date=today)
@pytest.mark.freeze_time("2022-12-14 15:00:00+01:00")
async def test_gas_model(aresponses: ResponsesMockServer) -> None:
"""Test the gas model - easyEnergy at 15:00:00 CET."""
aresponses.add(
"mijn.easyenergy.com",
"/nl/api/tariff/getlebatariffs",
"GET",
aresponses.Response(
status=200,
headers={"Content-Type": "application/json"},
text=load_fixtures("gas.json"),
),
)
async with ClientSession() as session:
today = date(2022, 12, 14)
client = EasyEnergy(session=session)
gas: Gas = await client.gas_prices(
start_date=today,
end_date=today,
vat=VatOption.INCLUDE,
)
assert gas is not None
assert isinstance(gas, Gas)
assert gas.extreme_prices[1] == 1.48534
assert gas.extreme_prices[0] == 1.4645
assert gas.average_price == 1.47951
assert gas.current_price == 1.48534
assert isinstance(gas.timestamp_prices, list)
@pytest.mark.freeze_time("2022-12-14 04:00:00+01:00")
async def test_gas_morning_model(aresponses: ResponsesMockServer) -> None:
"""Test the gas model in the morning - easyEnergy at 04:00:00 CET."""
aresponses.add(
"mijn.easyenergy.com",
"/nl/api/tariff/getlebatariffs",
"GET",
aresponses.Response(
status=200,
headers={"Content-Type": "application/json"},
text=load_fixtures("gas.json"),
),
)
async with ClientSession() as session:
today = date(2022, 12, 14)
client = EasyEnergy(session=session)
gas: Gas = await client.gas_prices(
start_date=today,
end_date=today,
vat=VatOption.INCLUDE,
)
assert gas is not None
assert isinstance(gas, Gas)
assert gas.extreme_prices[1] == 1.48534
assert gas.extreme_prices[0] == 1.4645
assert gas.average_price == 1.47951
assert gas.current_price == 1.4645
async def test_gas_none_data(aresponses: ResponsesMockServer) -> None:
"""Test when there is no data for the current datetime."""
aresponses.add(
"mijn.easyenergy.com",
"/nl/api/tariff/getlebatariffs",
"GET",
aresponses.Response(
status=200,
headers={"Content-Type": "application/json"},
text=load_fixtures("gas.json"),
),
)
async with ClientSession() as session:
today = date(2022, 12, 14)
client = EasyEnergy(session=session)
gas: Gas = await client.gas_prices(start_date=today, end_date=today)
assert gas is not None
assert isinstance(gas, Gas)
assert gas.current_price is None
assert gas.average_price == 1.47951
async def test_no_gas_data(aresponses: ResponsesMockServer) -> None:
"""Test when there is no gas data."""
aresponses.add(
"mijn.easyenergy.com",
"/nl/api/tariff/getlebatariffs",
"GET",
aresponses.Response(
status=200,
headers={"Content-Type": "application/json"},
text="[]",
),
)
async with ClientSession() as session:
today = date(2022, 12, 16)
client = EasyEnergy(session=session)
with pytest.raises(EasyEnergyNoDataError):
await client.gas_prices(start_date=today, end_date=today)
|
