Skip to content

Commit 2deed47

Browse files
committed
Refactor eventloop for backtesting
1 parent b2bc3e3 commit 2deed47

18 files changed

+1152
-238
lines changed

tests/app/backtesting/test_strategy_saving.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -100,9 +100,6 @@ def test_in_memory(self):
100100
backtest_report_dir, report_name
101101
)
102102

103-
for root, dirs, files in os.walk(backtest_report_dir):
104-
for name in files:
105-
print(name)
106103
# check if the report json file exists
107104
self.assertTrue(os.path.isfile(report_file_path))
108105

tests/app/test_add_data_provider.py

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
from datetime import datetime
2+
3+
from investing_algorithm_framework import DataProvider, DataSource
4+
from tests.resources import TestBase
5+
6+
class DataProviderTest(DataProvider):
7+
8+
def has_data(self, data_source: DataSource, start_date: datetime = None,
9+
end_date: datetime = None) -> bool:
10+
return True
11+
12+
def get_data(self, data_source: DataSource, date: datetime = None,
13+
start_date: datetime = None, end_date: datetime = None,
14+
save: bool = False):
15+
return None
16+
17+
def prepare_backtest_data(self, data_source: DataSource,
18+
backtest_start_date, backtest_end_date) -> None:
19+
pass
20+
21+
def get_backtest_data(self, data_source: DataSource,
22+
backtest_index_date: datetime,
23+
backtest_start_date: datetime = None,
24+
backtest_end_date: datetime = None) -> None:
25+
return None
26+
27+
28+
class Test(TestBase):
29+
30+
def test_add(self):
31+
self.app.add_data_provider(DataProviderTest)
32+
self.app.initialize_data_sources()
33+
self.assertEqual(1, self.app.container.data_provider_service.count())

0 commit comments

Comments
 (0)