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