pikers
/
piker
3
0
Fork 0
Code Issues 4 Pull Requests 6 Packages Projects Releases Wiki Activity

Ensure not to write to pps when asserting?

paper_trade_improvements_rebase
jaredgoldman 2023-02-24 17:34:58 -05:00
parent bf1757de07
commit 4d32c6f6a2
1 changed files with 23 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
tests/test_paper.py
View File

@ -137,17 +137,34 @@ def test_paper_trade(open_test_pikerd: AsyncContextManager):
# Close piker like a user would # Close piker like a user would
raise KeyboardInterrupt raise KeyboardInterrupt
# def _assert_entries():
# cleared_ledger_entry = {}
# with open_trade_ledger(broker, test_account) as ledger:
# cleared_ledger_entry = ledger[oid]
# assert list(ledger.keys())[-1] == oid
# assert cleared_ledger_entry["size"] == test_size
# assert cleared_ledger_entry["fqsn"] == fqsn
#
# with open_pps(broker, test_pp_account, False) as table:
# pp_price = table.conf[broker][test_pp_account][fqsn]["ppu"]
# # Ensure the price-per-unit (breakeven) price is close to our clearing price
# assert math.isclose(
# pp_price, cleared_ledger_entry["size"], rel_tol=1
# )
# assert table.brokername == broker
# assert table.acctid == test_pp_account
#
# Open piker load pps locally # Open piker load pps locally
# and ensure last pps price is the same as ledger entry # and ensure last pps price is the same as ledger entry
def _assert_pps(ledger, table): def _assert_pps(ledger, table):
return ( assert positions[(broker, test_account)][-1]["avg_price"] == ledger[oid]["price"]
positions[(broker, test_account)][-1]["avg_price"] == ledger[oid]["price"]
)
def _assert_no_pps(ledger, table): def _assert_no_pps(ledger, table):
print(f"positions: {positions}") print(f"positions: {positions}")
return not bool(table) assert not bool(table)
# return len(table.pps) == 0 # return len(table.pps) == 0
# Close position and assert empty position in pps # Close position and assert empty position in pps
@ -160,7 +177,7 @@ def test_paper_trade(open_test_pikerd: AsyncContextManager):
open_pps(broker, test_pp_account) as table, open_pps(broker, test_pp_account) as table,
): ):
if assert_cb: if assert_cb:
assert assert_cb(ledger, table) assert_cb(ledger, table)
for exception in exc_info.value.exceptions: for exception in exc_info.value.exceptions:
assert isinstance(exception, KeyboardInterrupt) or isinstance( assert isinstance(exception, KeyboardInterrupt) or isinstance(
@ -175,6 +192,7 @@ def test_paper_trade(open_test_pikerd: AsyncContextManager):
open_pikerd=open_test_pikerd, open_pikerd=open_test_pikerd,
action="buy", action="buy",
), ),
# _assert_entries
) )
_run_test_and_check( _run_test_and_check(