|
15 | 15 | Client, |
16 | 16 | Currency, |
17 | 17 | DeliveryArea, |
| 18 | + DeliveryDuration, |
18 | 19 | DeliveryPeriod, |
19 | 20 | EnergyMarketCodeType, |
20 | 21 | MarketSide, |
@@ -57,7 +58,7 @@ def __init__(self) -> None: |
57 | 58 | ) |
58 | 59 | self.delivery_period = DeliveryPeriod( |
59 | 60 | start=self.delivery_start, |
60 | | - duration=timedelta(minutes=15), |
| 61 | + duration=DeliveryDuration.from_timedelta(timedelta(minutes=15)), |
61 | 62 | ) |
62 | 63 | self.price = Price(amount=Decimal("56"), currency=Currency.EUR) |
63 | 64 | self.quantity = Power(mw=Decimal("0.1")) |
@@ -134,7 +135,7 @@ async def create_test_trade( |
134 | 135 | ) |
135 | 136 | delivery_period = DeliveryPeriod( |
136 | 137 | start=delivery_start, |
137 | | - duration=timedelta(minutes=15), |
| 138 | + duration=DeliveryDuration.from_timedelta(timedelta(minutes=15)), |
138 | 139 | ) |
139 | 140 | buy_order = await create_test_order( |
140 | 141 | trader=trader, |
@@ -235,9 +236,7 @@ async def test_list_gridpool_orders(trader: _TestTrader) -> None: |
235 | 236 | # filter by delivery period to avoid fetching too many orders |
236 | 237 | orders = [ |
237 | 238 | order |
238 | | - async for order in trader.client.list_gridpool_orders( |
239 | | - gridpool_id=GRIDPOOL_ID, delivery_period=trader.delivery_period |
240 | | - ) |
| 239 | + async for order in trader.client.list_gridpool_orders(gridpool_id=GRIDPOOL_ID) |
241 | 240 | ] |
242 | 241 | listed_orders_id = [order.order_id for order in orders] |
243 | 242 | for order_id in created_orders_id: |
@@ -350,14 +349,8 @@ async def test_cancel_all_orders(trader: _TestTrader) -> None: |
350 | 349 |
|
351 | 350 | async def test_list_gridpool_trades(trader: _TestTrader) -> None: |
352 | 351 | """Test listing gridpool trades.""" |
353 | | - buy_order, sell_order = await create_test_trade(trader) |
354 | | - trades = [ |
355 | | - trade |
356 | | - async for trade in trader.client.list_gridpool_trades( |
357 | | - GRIDPOOL_ID, |
358 | | - delivery_period=buy_order.order.delivery_period, |
359 | | - ) |
360 | | - ] |
| 352 | + await create_test_trade(trader) |
| 353 | + trades = [trade async for trade in trader.client.list_gridpool_trades(GRIDPOOL_ID)] |
361 | 354 | assert len(trades) >= 1 |
362 | 355 |
|
363 | 356 |
|
@@ -397,7 +390,10 @@ async def test_receive_public_trades_filter(trader: _TestTrader) -> None: |
397 | 390 | start_time = datetime.now(timezone.utc).replace(second=0, microsecond=0) |
398 | 391 | start_time += timedelta(minutes=30 - start_time.minute % 15) # next 15-minute mark |
399 | 392 |
|
400 | | - delivery_period = DeliveryPeriod(start=start_time, duration=timedelta(minutes=15)) |
| 393 | + delivery_period = DeliveryPeriod( |
| 394 | + start=start_time, |
| 395 | + duration=DeliveryDuration.from_timedelta(timedelta(minutes=15)), |
| 396 | + ) |
401 | 397 |
|
402 | 398 | price = Price(amount=Decimal("808"), currency=Currency.EUR) |
403 | 399 |
|
|
0 commit comments