From 62f7606d2c025e793aa24bd041a9d50a07bd0748 Mon Sep 17 00:00:00 2001 From: Matthias Date: Sat, 23 Jul 2022 16:05:25 +0200 Subject: [PATCH] Update tests to new variant --- tests/freqai/conftest.py | 2 +- tests/freqai/test_freqai_datakitchen.py | 8 ++++---- tests/freqai/test_freqai_interface.py | 8 ++++---- 3 files changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/freqai/conftest.py b/tests/freqai/conftest.py index 549ba2663..1d98fd863 100644 --- a/tests/freqai/conftest.py +++ b/tests/freqai/conftest.py @@ -61,7 +61,7 @@ def get_patched_data_kitchen(mocker, freqaiconf): def get_patched_freqai_strategy(mocker, freqaiconf): strategy = StrategyResolver.load_strategy(freqaiconf) - strategy.bot_start() + strategy.ft_bot_start() return strategy diff --git a/tests/freqai/test_freqai_datakitchen.py b/tests/freqai/test_freqai_datakitchen.py index 1964d1423..d0d82b489 100644 --- a/tests/freqai/test_freqai_datakitchen.py +++ b/tests/freqai/test_freqai_datakitchen.py @@ -69,7 +69,7 @@ def test_update_historic_data(mocker, default_conf): strategy = get_patched_freqai_strategy(mocker, freqaiconf) exchange = get_patched_exchange(mocker, freqaiconf) strategy.dp = DataProvider(freqaiconf, exchange) - freqai = strategy.model.bridge + freqai = strategy.freqai freqai.live = True freqai.dk = FreqaiDataKitchen(freqaiconf, freqai.dd) timerange = TimeRange.parse_timerange("20180110-20180114") @@ -105,7 +105,7 @@ def test_load_all_pairs_histories(mocker, default_conf): strategy = get_patched_freqai_strategy(mocker, freqaiconf) exchange = get_patched_exchange(mocker, freqaiconf) strategy.dp = DataProvider(freqaiconf, exchange) - freqai = strategy.model.bridge + freqai = strategy.freqai freqai.live = True freqai.dk = FreqaiDataKitchen(freqaiconf, freqai.dd) timerange = TimeRange.parse_timerange("20180110-20180114") @@ -125,7 +125,7 @@ def test_get_base_and_corr_dataframes(mocker, default_conf): strategy = get_patched_freqai_strategy(mocker, freqaiconf) exchange = get_patched_exchange(mocker, freqaiconf) strategy.dp = DataProvider(freqaiconf, exchange) - freqai = strategy.model.bridge + freqai = strategy.freqai freqai.live = True freqai.dk = FreqaiDataKitchen(freqaiconf, freqai.dd) timerange = TimeRange.parse_timerange("20180110-20180114") @@ -153,7 +153,7 @@ def test_use_strategy_to_populate_indicators(mocker, default_conf): exchange = get_patched_exchange(mocker, freqaiconf) strategy.dp = DataProvider(freqaiconf, exchange) strategy.freqai_info = freqaiconf.get("freqai", {}) - freqai = strategy.model.bridge + freqai = strategy.freqai freqai.live = True freqai.dk = FreqaiDataKitchen(freqaiconf, freqai.dd) timerange = TimeRange.parse_timerange("20180110-20180114") diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py index 9219baee3..e812dd47e 100644 --- a/tests/freqai/test_freqai_interface.py +++ b/tests/freqai/test_freqai_interface.py @@ -21,7 +21,7 @@ def test_train_model_in_series_LightGBM(mocker, default_conf): exchange = get_patched_exchange(mocker, freqaiconf) strategy.dp = DataProvider(freqaiconf, exchange) strategy.freqai_info = freqaiconf.get("freqai", {}) - freqai = strategy.model.bridge + freqai = strategy.freqai freqai.live = True freqai.dk = FreqaiDataKitchen(freqaiconf, freqai.dd) timerange = TimeRange.parse_timerange("20180110-20180130") @@ -113,7 +113,7 @@ def test_start_backtesting(mocker, default_conf): exchange = get_patched_exchange(mocker, freqaiconf) strategy.dp = DataProvider(freqaiconf, exchange) strategy.freqai_info = freqaiconf.get("freqai", {}) - freqai = strategy.model.bridge + freqai = strategy.freqai freqai.live = False freqai.dk = FreqaiDataKitchen(freqaiconf, freqai.dd) timerange = TimeRange.parse_timerange("20180110-20180130") @@ -139,7 +139,7 @@ def test_start_backtesting_from_existing_folder(mocker, default_conf, caplog): exchange = get_patched_exchange(mocker, freqaiconf) strategy.dp = DataProvider(freqaiconf, exchange) strategy.freqai_info = freqaiconf.get("freqai", {}) - freqai = strategy.model.bridge + freqai = strategy.freqai freqai.live = False freqai.dk = FreqaiDataKitchen(freqaiconf, freqai.dd) timerange = TimeRange.parse_timerange("20180110-20180130") @@ -162,7 +162,7 @@ def test_start_backtesting_from_existing_folder(mocker, default_conf, caplog): exchange = get_patched_exchange(mocker, freqaiconf) strategy.dp = DataProvider(freqaiconf, exchange) strategy.freqai_info = freqaiconf.get("freqai", {}) - freqai = strategy.model.bridge + freqai = strategy.freqai freqai.live = False freqai.dk = FreqaiDataKitchen(freqaiconf, freqai.dd) timerange = TimeRange.parse_timerange("20180110-20180130")