separated test_leverage into test_interest and test_liquidation_price, and paramaterized tests

This commit is contained in:
Sam Germain
2021-08-25 14:27:55 -06:00
parent a087d03db9
commit fe5e00361e
3 changed files with 115 additions and 96 deletions

View File

@@ -13,12 +13,7 @@ def liquidation_price(
collateral: Optional[Collateral]
) -> Optional[float]:
leverage_exchanges = [
'binance',
'kraken',
'ftx'
]
if trading_mode == TradingMode.SPOT or exchange_name.lower() not in leverage_exchanges:
if trading_mode == TradingMode.SPOT:
return None
if not collateral:
@@ -34,7 +29,7 @@ def liquidation_price(
elif exchange_name.lower() == "ftx":
return ftx(open_rate, is_short, leverage, trading_mode, collateral)
raise OperationalException(
f"liquidation_price is not yet implemented for {exchange_name}"
f"liquidation_price is not implemented for {exchange_name}"
)