@@ -213,6 +213,8 @@ async def test_case_1(
213
213
await bounds_rx .receive (), power = 1000.0 , lower = - 4000.0 , upper = 4000.0
214
214
)
215
215
216
+ await asyncio .sleep (0.0 )
217
+
216
218
assert set_power .call_count == 4
217
219
assert sorted (set_power .call_args_list ) == [
218
220
mocker .call (inv_id , 250.0 )
@@ -248,6 +250,7 @@ async def side_effect(inv_id: int, _: float) -> None:
248
250
result , power_distributing .Success
249
251
),
250
252
)
253
+ await asyncio .sleep (0.0 )
251
254
assert set_power .call_count == 4
252
255
assert sorted (set_power .call_args_list ) == [
253
256
mocker .call (inv_id , 25.0 ) for inv_id in mocks .microgrid .battery_inverter_ids
@@ -267,6 +270,7 @@ async def side_effect(inv_id: int, _: float) -> None:
267
270
268
271
# There should be an automatic retry.
269
272
set_power .side_effect = None
273
+ await asyncio .sleep (0.0 )
270
274
assert set_power .call_count == 4
271
275
assert sorted (set_power .call_args_list ) == [
272
276
mocker .call (inv_id , 25.0 ) for inv_id in mocks .microgrid .battery_inverter_ids
@@ -318,6 +322,7 @@ async def test_case_2(self, mocks: Mocks, mocker: MockerFixture) -> None:
318
322
self ._assert_report (
319
323
await bounds_1_rx .receive (), power = 1000.0 , lower = - 2000.0 , upper = 2000.0
320
324
)
325
+ await asyncio .sleep (0.0 )
321
326
assert set_power .call_count == 2
322
327
assert sorted (set_power .call_args_list ) == [
323
328
mocker .call (inv_id , 500.0 )
@@ -331,6 +336,7 @@ async def test_case_2(self, mocks: Mocks, mocker: MockerFixture) -> None:
331
336
if not latest_dist_result_2 .has_value ():
332
337
bounds = await bounds_2_rx .receive ()
333
338
self ._assert_report (bounds , power = 1000.0 , lower = - 2000.0 , upper = 2000.0 )
339
+ await asyncio .sleep (0.0 )
334
340
assert set_power .call_count == 2
335
341
assert sorted (set_power .call_args_list ) == [
336
342
mocker .call (inv_id , 500.0 )
@@ -376,6 +382,7 @@ async def test_case_3(self, mocks: Mocks, mocker: MockerFixture) -> None:
376
382
await bounds_2_rx .receive (), power = - 1000.0 , lower = - 1000.0 , upper = 0.0
377
383
)
378
384
385
+ await asyncio .sleep (0.0 )
379
386
assert set_power .call_count == 4
380
387
assert sorted (set_power .call_args_list ) == [
381
388
mocker .call (inv_id , - 250.0 )
@@ -395,6 +402,7 @@ async def test_case_3(self, mocks: Mocks, mocker: MockerFixture) -> None:
395
402
bounds = await bounds_2_rx .receive ()
396
403
self ._assert_report (bounds , power = 0.0 , lower = - 1000.0 , upper = 0.0 )
397
404
405
+ await asyncio .sleep (0.0 )
398
406
assert set_power .call_count == 4
399
407
assert sorted (set_power .call_args_list ) == [
400
408
mocker .call (inv_id , 0.0 ) for inv_id in mocks .microgrid .battery_inverter_ids
@@ -428,6 +436,7 @@ async def test_case_4(self, mocks: Mocks, mocker: MockerFixture) -> None:
428
436
self ._assert_report (
429
437
await bounds_rx .receive (), power = 1000.0 , lower = - 4000.0 , upper = 4000.0
430
438
)
439
+ await asyncio .sleep (0.0 )
431
440
assert set_power .call_count == 4
432
441
assert sorted (set_power .call_args_list ) == [
433
442
mocker .call (inv_id , 250.0 )
@@ -453,6 +462,7 @@ async def test_case_4(self, mocks: Mocks, mocker: MockerFixture) -> None:
453
462
self ._assert_report (
454
463
await bounds_rx .receive (), power = 400.0 , lower = - 4000.0 , upper = 4000.0
455
464
)
465
+ await asyncio .sleep (0.0 )
456
466
assert set_power .call_count == 4
457
467
assert sorted (set_power .call_args_list ) == [
458
468
mocker .call (inv_id , 100.0 )
@@ -477,6 +487,7 @@ async def test_case_4(self, mocks: Mocks, mocker: MockerFixture) -> None:
477
487
self ._assert_report (
478
488
await bounds_rx .receive (), power = 0.0 , lower = - 4000.0 , upper = 4000.0
479
489
)
490
+ await asyncio .sleep (0.0 )
480
491
assert set_power .call_count == 4
481
492
assert sorted (set_power .call_args_list ) == [
482
493
mocker .call (inv_id , 0.0 ) for inv_id in mocks .microgrid .battery_inverter_ids
@@ -501,6 +512,7 @@ async def test_case_4(self, mocks: Mocks, mocker: MockerFixture) -> None:
501
512
self ._assert_report (
502
513
await bounds_rx .receive (), power = - 400.0 , lower = - 4000.0 , upper = 4000.0
503
514
)
515
+ await asyncio .sleep (0.0 )
504
516
assert set_power .call_count == 4
505
517
assert sorted (set_power .call_args_list ) == [
506
518
mocker .call (inv_id , - 100.0 )
@@ -587,6 +599,7 @@ async def test_case_5( # pylint: disable=too-many-statements,too-many-locals
587
599
await bounds_1_rx .receive (), power = 200.0 , lower = - 1000.0 , upper = 1500.0
588
600
)
589
601
602
+ await asyncio .sleep (0.0 )
590
603
assert set_power .call_count == 4
591
604
assert sorted (set_power .call_args_list ) == [
592
605
mocker .call (inv_id , 50.0 ) for inv_id in mocks .microgrid .battery_inverter_ids
@@ -624,6 +637,7 @@ async def test_case_5( # pylint: disable=too-many-statements,too-many-locals
624
637
if dist_result .succeeded_power == Power .from_watts (720.0 ):
625
638
break
626
639
640
+ await asyncio .sleep (0.0 )
627
641
assert set_power .call_count == 4
628
642
assert sorted (set_power .call_args_list ) == [
629
643
mocker .call (inv_id , 720.0 / 4 )
@@ -664,6 +678,7 @@ async def test_case_5( # pylint: disable=too-many-statements,too-many-locals
664
678
if dist_result .succeeded_power == Power .from_watts (- 280.0 ):
665
679
break
666
680
681
+ await asyncio .sleep (0.0 )
667
682
assert set_power .call_count == 4
668
683
assert sorted (set_power .call_args_list ) == [
669
684
mocker .call (inv_id , - 280.0 / 4 )
0 commit comments