Minor reformatting
parent
b180602a3e
commit
dff8abd6ad
|
@ -9,7 +9,6 @@ from piker.clearing._messages import (
|
||||||
from uuid import uuid4
|
from uuid import uuid4
|
||||||
from typing import (
|
from typing import (
|
||||||
AsyncContextManager,
|
AsyncContextManager,
|
||||||
Any,
|
|
||||||
Literal,
|
Literal,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -17,7 +16,6 @@ from functools import partial
|
||||||
from piker.pp import (
|
from piker.pp import (
|
||||||
open_trade_ledger,
|
open_trade_ledger,
|
||||||
open_pps,
|
open_pps,
|
||||||
PpTable
|
|
||||||
)
|
)
|
||||||
|
|
||||||
from piker.clearing import (
|
from piker.clearing import (
|
||||||
|
@ -39,7 +37,6 @@ def test_paper_trade(
|
||||||
open_test_pikerd: AsyncContextManager
|
open_test_pikerd: AsyncContextManager
|
||||||
):
|
):
|
||||||
|
|
||||||
cleared_price: float
|
|
||||||
test_exec_mode='live'
|
test_exec_mode='live'
|
||||||
test_account = 'paper'
|
test_account = 'paper'
|
||||||
test_size = 1
|
test_size = 1
|
||||||
|
@ -53,13 +50,20 @@ def test_paper_trade(
|
||||||
tuple[str, str],
|
tuple[str, str],
|
||||||
list[BrokerdPosition],
|
list[BrokerdPosition],
|
||||||
]
|
]
|
||||||
|
cleared_price: float
|
||||||
|
|
||||||
|
|
||||||
async def _async_main(
|
async def _async_main(
|
||||||
open_pikerd: AsyncContextManager,
|
open_pikerd: AsyncContextManager,
|
||||||
action: Literal['buy', 'sell'] | None = None,
|
action: Literal['buy', 'sell'] | None = None,
|
||||||
price: int = 30000,
|
price: int = 30000,
|
||||||
assert_entries: bool = False,
|
assert_entries: bool = False,
|
||||||
) -> Any:
|
) -> None:
|
||||||
|
"""Spawn a paper piper actor, place a trade and assert entries are present
|
||||||
|
in both trade ledger and pps tomls. Then restart piker and ensure
|
||||||
|
that pps from previous trade exists in the ems pps.
|
||||||
|
Finally close the position and ensure that the position in pps.toml is closed.
|
||||||
|
"""
|
||||||
|
|
||||||
oid: str = str(uuid4())
|
oid: str = str(uuid4())
|
||||||
book: OrderBook
|
book: OrderBook
|
||||||
|
@ -134,7 +138,7 @@ def test_paper_trade(
|
||||||
# run initial time and send sent and assert trade
|
# run initial time and send sent and assert trade
|
||||||
with pytest.raises(
|
with pytest.raises(
|
||||||
BaseExceptionGroup
|
BaseExceptionGroup
|
||||||
) as exc_info:
|
):
|
||||||
trio.run(partial(_async_main,
|
trio.run(partial(_async_main,
|
||||||
open_pikerd=open_test_pikerd,
|
open_pikerd=open_test_pikerd,
|
||||||
action='buy',
|
action='buy',
|
||||||
|
@ -143,11 +147,11 @@ def test_paper_trade(
|
||||||
|
|
||||||
with pytest.raises(
|
with pytest.raises(
|
||||||
BaseExceptionGroup
|
BaseExceptionGroup
|
||||||
) as exc_info:
|
):
|
||||||
trio.run(_open_and_assert_pps)
|
trio.run(_open_and_assert_pps)
|
||||||
|
|
||||||
with pytest.raises(
|
with pytest.raises(
|
||||||
BaseExceptionGroup
|
BaseExceptionGroup
|
||||||
) as exc_info:
|
):
|
||||||
trio.run(_close_pp_and_assert)
|
trio.run(_close_pp_and_assert)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue