adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix filename handling with --strategy-list

Browse Source
This commit is contained in:
Matthias 2020-04-02 17:29:18 +02:00
parent 2915917680
commit cf6e6488c7
2 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
freqtrade/optimize/optimize_reports.py
View File

@ -24,13 +24,14 @@ def store_backtest_result(recordfilename: Path, all_results: Dict[str, DataFrame
for index, t in results.iterrows()] for index, t in results.iterrows()]
if records: if records:
filename = recordfilename
if len(all_results) > 1: if len(all_results) > 1:
# Inject strategy to filename # Inject strategy to filename
recordfilename = Path.joinpath( filename = Path.joinpath(
recordfilename.parent, recordfilename.parent,
f'{recordfilename.stem}-{strategy}').with_suffix(recordfilename.suffix) f'{recordfilename.stem}-{strategy}').with_suffix(recordfilename.suffix)
logger.info(f'Dumping backtest results to {recordfilename}') logger.info(f'Dumping backtest results to {filename}')
file_dump_json(recordfilename, records) file_dump_json(filename, records)
def generate_text_table(data: Dict[str, Dict], stake_currency: str, max_open_trades: int, def generate_text_table(data: Dict[str, Dict], stake_currency: str, max_open_trades: int,

9
tests/optimize/test_optimize_reports.py
View File

@ -160,10 +160,15 @@ def test_backtest_record(default_conf, fee, mocker):
# reset test to test with strategy name # reset test to test with strategy name
names = [] names = []
records = [] records = []
results['Strat'] = pd.DataFrame() results['Strat'] = results['DefStrat']
results['Strat2'] = results['DefStrat']
store_backtest_result(Path("backtest-result.json"), results) store_backtest_result(Path("backtest-result.json"), results)
# Assert file_dump_json was only called once # Assert file_dump_json was only called once
assert names == [Path('backtest-result-DefStrat.json')] assert names == [
Path('backtest-result-DefStrat.json'),
Path('backtest-result-Strat.json'),
Path('backtest-result-Strat2.json'),
]
records = records[0] records = records[0]
# Ensure records are of correct type # Ensure records are of correct type
assert len(records) == 4 assert len(records) == 4