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
|
"""Tests for aiopvpc."""
from datetime import datetime
from typing import cast
import pytest
from aiopvpc.const import DataSource, KEY_PVPC, REFERENCE_TZ, TARIFFS, UTC_TZ
from aiopvpc.pvpc_data import PVPCData
from tests.conftest import MockAsyncSession, TZ_TEST
@pytest.mark.parametrize(
"local_tz, source, tariff, expected_18h",
(
(TZ_TEST, "esios_public", TARIFFS[0], 0.23144),
(REFERENCE_TZ, "esios_public", TARIFFS[0], 0.23144),
(REFERENCE_TZ, "esios_public", TARIFFS[1], 0.13813),
(REFERENCE_TZ, "esios", TARIFFS[0], 0.23144),
(REFERENCE_TZ, "esios", TARIFFS[1], 0.13813),
(TZ_TEST, "esios", TARIFFS[0], 0.23144),
),
)
@pytest.mark.asyncio
async def test_geo_ids(local_tz, source, tariff, expected_18h):
"""Test different prices for different geo zones."""
start = datetime(2021, 6, 1, 10, tzinfo=UTC_TZ)
mock_session = MockAsyncSession()
pvpc_data = PVPCData(
session=mock_session,
tariff=tariff,
local_timezone=local_tz,
data_source=cast(DataSource, source),
api_token="test-token" if source == "esios" else None,
)
api_data = await pvpc_data.async_update_all(None, start)
assert all(api_data.availability.values())
assert pvpc_data.process_state_and_attributes(api_data, KEY_PVPC, start)
# for ts, price in pvpc_data._current_prices.items():
# print(f"{ts.astimezone(local_tz):%H}h --> {price:.5f} ")
ts_loc_18h_utc = datetime(2021, 6, 1, 18, tzinfo=local_tz).astimezone(UTC_TZ)
price_loc_18h = api_data.sensors[KEY_PVPC][ts_loc_18h_utc]
assert price_loc_18h == expected_18h
|
