Switch combined profit response to dataclass

This commit is contained in:
Matthias
2023-08-29 20:47:06 +02:00
parent c58a1649cb
commit 28e685ee2b
2 changed files with 26 additions and 17 deletions

View File

@@ -1159,21 +1159,21 @@ def test_calc_profit(
trade.open_order_id = 'something'
profit_res = trade.calc_profit_combined(close_rate)
assert pytest.approx(profit_res['profit_abs']) == round(profit, 8)
assert pytest.approx(profit_res['profit_ratio']) == round(profit_ratio, 8)
assert pytest.approx(profit_res.profit_abs) == round(profit, 8)
assert pytest.approx(profit_res.profit_ratio) == round(profit_ratio, 8)
assert pytest.approx(profit_res['total_profit']) == round(profit, 8)
# assert pytest.approx(profit_res['total_profit_ratio']) == round(profit_ratio, 8)
assert pytest.approx(profit_res.total_profit) == round(profit, 8)
# assert pytest.approx(profit_res.total_profit_ratio) == round(profit_ratio, 8)
assert pytest.approx(trade.calc_profit(rate=close_rate)) == round(profit, 8)
assert pytest.approx(trade.calc_profit_ratio(rate=close_rate)) == round(profit_ratio, 8)
profit_res2 = trade.calc_profit_combined(close_rate, trade.amount, trade.open_rate)
assert pytest.approx(profit_res2['profit_abs']) == round(profit, 8)
assert pytest.approx(profit_res2['profit_ratio']) == round(profit_ratio, 8)
assert pytest.approx(profit_res2.profit_abs) == round(profit, 8)
assert pytest.approx(profit_res2.profit_ratio) == round(profit_ratio, 8)
assert pytest.approx(profit_res2['total_profit']) == round(profit, 8)
# assert pytest.approx(profit_res2['total_profit_ratio']) == round(profit_ratio, 8)
assert pytest.approx(profit_res2.total_profit) == round(profit, 8)
# assert pytest.approx(profit_res2.total_profit_ratio) == round(profit_ratio, 8)
assert pytest.approx(trade.calc_profit(close_rate, trade.amount,
trade.open_rate)) == round(profit, 8)