Add market data handling, battery state updates, and unit tests

This commit is contained in:
2026-09-22 15:43:30 +01:00
parent 34a7aeaf03
commit 4e37339534
5 changed files with 139 additions and 2 deletions
+16
View File
@@ -0,0 +1,16 @@
import datetime
from batterymodel.models.market import MarketState
def test_market():
this_market = MarketState()
this_market._market_data[datetime.datetime.strptime('01-01-2022 00:00', '%d-%m-%Y %H:%M')] = 11
this_market._market_data[datetime.datetime.strptime('01-01-2022 00:01', '%d-%m-%Y %H:%M')] = 12
this_market._market_data[datetime.datetime.strptime('01-01-2022 00:02', '%d-%m-%Y %H:%M')] = 13
this_market._market_data[datetime.datetime.strptime('01-01-2022 00:05', '%d-%m-%Y %H:%M')] = 14
assert this_market.get_market_rate_for_timestamp(datetime.datetime.strptime('01-01-2022 00:02', '%d-%m-%Y %H:%M')) == 13
assert this_market.get_market_rate_for_timestamp(datetime.datetime.strptime('01-01-2022 00:03', '%d-%m-%Y %H:%M')) == 13