From 8085e24dcde19744e5b8e42ff9ce8212a21c2d88 Mon Sep 17 00:00:00 2001 From: jainanuj94 Date: Tue, 6 Aug 2024 20:00:05 +0530 Subject: [PATCH] update tests --- tests/optimize/test_optimize_reports.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/optimize/test_optimize_reports.py b/tests/optimize/test_optimize_reports.py index abbed1e54..40673a1b6 100644 --- a/tests/optimize/test_optimize_reports.py +++ b/tests/optimize/test_optimize_reports.py @@ -298,6 +298,8 @@ def test_store_backtest_candles(testdatadir, mocker): assert dump_mock.call_count == 3 assert isinstance(dump_mock.call_args_list[0][0][0], Path) assert str(dump_mock.call_args_list[0][0][0]).endswith("_signals.pkl") + assert str(dump_mock.call_args_list[1][0][0]).endswith("_rejected.pkl") + assert str(dump_mock.call_args_list[2][0][0]).endswith("_exited.pkl") dump_mock.reset_mock() # mock file exporting @@ -307,6 +309,9 @@ def test_store_backtest_candles(testdatadir, mocker): assert isinstance(dump_mock.call_args_list[0][0][0], Path) # result will be testdatadir / testresult-_signals.pkl assert str(dump_mock.call_args_list[0][0][0]).endswith("_signals.pkl") + assert str(dump_mock.call_args_list[1][0][0]).endswith("_rejected.pkl") + assert str(dump_mock.call_args_list[2][0][0]).endswith("_exited.pkl") + dump_mock.reset_mock()