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

Reenable Catboost test

Browse Source
This commit is contained in:
Matthias 2022-07-23 17:09:29 +02:00
parent 81c1aa3c13
commit b149eaf71b
1 changed files with 45 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

87
tests/freqai/test_freqai_interface.py
View File

@ -1,11 +1,13 @@
# from unittest.mock import MagicMock # from unittest.mock import MagicMock
# from freqtrade.commands.optimize_commands import setup_optimize_configuration, start_edge # from freqtrade.commands.optimize_commands import setup_optimize_configuration, start_edge
import copy import copy
# import platform import platform
import shutil import shutil
from pathlib import Path from pathlib import Path
from unittest.mock import MagicMock from unittest.mock import MagicMock
import pytest
from freqtrade.configuration import TimeRange from freqtrade.configuration import TimeRange
from freqtrade.data.dataprovider import DataProvider from freqtrade.data.dataprovider import DataProvider
from freqtrade.freqai.data_kitchen import FreqaiDataKitchen from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
@ -58,52 +60,53 @@ def test_train_model_in_series_LightGBM(mocker, default_conf):
shutil.rmtree(Path(freqai.dk.full_path)) shutil.rmtree(Path(freqai.dk.full_path))
# FIXME: hits segfault @pytest.mark.skipif("arm" in platform.uname()[-1], reason="no ARM for Catboost ...")
# @pytest.mark.skipif("arm" in platform.uname()[-1], reason="no ARM..") def test_train_model_in_series_Catboost(mocker, default_conf):
# def test_train_model_in_series_Catboost(mocker, default_conf): freqaiconf = freqai_conf(copy.deepcopy(default_conf))
# freqaiconf = freqai_conf(copy.deepcopy(default_conf)) freqaiconf.update({"timerange": "20180110-20180130"})
# freqaiconf.update({"timerange": "20180110-20180130"}) freqaiconf.update({"freqaimodel": "CatboostPredictionModel"})
# freqaiconf.update({"freqaimodel": "CatboostPredictionModel"}) del freqaiconf['freqai']['model_training_parameters']['verbosity']
# strategy = get_patched_freqai_strategy(mocker, freqaiconf) strategy = get_patched_freqai_strategy(mocker, freqaiconf)
# exchange = get_patched_exchange(mocker, freqaiconf) exchange = get_patched_exchange(mocker, freqaiconf)
# strategy.dp = DataProvider(freqaiconf, exchange) strategy.dp = DataProvider(freqaiconf, exchange)
# strategy.freqai_info = freqaiconf.get("freqai", {})
# freqai = strategy.model.bridge
# freqai.live = True
# freqai.dk = FreqaiDataKitchen(freqaiconf, freqai.dd)
# timerange = TimeRange.parse_timerange("20180110-20180130")
# freqai.dk.load_all_pair_histories(timerange)
# freqai.dd.pair_dict = MagicMock() strategy.freqai_info = freqaiconf.get("freqai", {})
freqai = strategy.freqai
freqai.live = True
freqai.dk = FreqaiDataKitchen(freqaiconf, freqai.dd)
timerange = TimeRange.parse_timerange("20180110-20180130")
freqai.dk.load_all_pair_histories(timerange)
# data_load_timerange = TimeRange.parse_timerange("20180110-20180130") freqai.dd.pair_dict = MagicMock()
# new_timerange = TimeRange.parse_timerange("20180120-20180130")
# freqai.train_model_in_series(new_timerange, "ADA/BTC", data_load_timerange = TimeRange.parse_timerange("20180110-20180130")
# strategy, freqai.dk, data_load_timerange) new_timerange = TimeRange.parse_timerange("20180120-20180130")
# assert ( freqai.train_model_in_series(new_timerange, "ADA/BTC",
# Path(freqai.dk.data_path / str(freqai.dk.model_filename + "_model.joblib")) strategy, freqai.dk, data_load_timerange)
# .resolve()
# .exists()
# )
# assert (
# Path(freqai.dk.data_path / str(freqai.dk.model_filename + "_metadata.json"))
# .resolve()
# .exists()
# )
# assert (
# Path(freqai.dk.data_path / str(freqai.dk.model_filename + "_trained_df.pkl"))
# .resolve()
# .exists()
# )
# assert (
# Path(freqai.dk.data_path / str(freqai.dk.model_filename + "_svm_model.joblib"))
# .resolve()
# .exists()
# )
# shutil.rmtree(Path(freqai.dk.full_path)) assert (
Path(freqai.dk.data_path / str(freqai.dk.model_filename + "_model.joblib"))
.resolve()
.exists()
)
assert (
Path(freqai.dk.data_path / str(freqai.dk.model_filename + "_metadata.json"))
.resolve()
.exists()
)
assert (
Path(freqai.dk.data_path / str(freqai.dk.model_filename + "_trained_df.pkl"))
.resolve()
.exists()
)
assert (
Path(freqai.dk.data_path / str(freqai.dk.model_filename + "_svm_model.joblib"))
.resolve()
.exists()
)
shutil.rmtree(Path(freqai.dk.full_path))
def test_start_backtesting(mocker, default_conf): def test_start_backtesting(mocker, default_conf):