lychiyu/freqtrade_origin
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2024-11-10 10:21:59 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Blackify

Browse Source
This commit is contained in:
Matthias 2019-01-22 06:56:54 +01:00
parent e66808bb02
commit 1e7431a7b8
1 changed files with 32 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
freqtrade/tests/data/test_dataprovider.py
View File

@ -8,74 +8,73 @@ from freqtrade.tests.conftest import get_patched_exchange
def test_ohlcv(mocker, default_conf, ticker_history): def test_ohlcv(mocker, default_conf, ticker_history):
default_conf['runmode'] = RunMode.DRY_RUN default_conf["runmode"] = RunMode.DRY_RUN
tick_interval = default_conf['ticker_interval'] tick_interval = default_conf["ticker_interval"]
exchange = get_patched_exchange(mocker, default_conf) exchange = get_patched_exchange(mocker, default_conf)
exchange._klines[('XRP/BTC', tick_interval)] = ticker_history exchange._klines[("XRP/BTC", tick_interval)] = ticker_history
exchange._klines[('UNITTEST/BTC', tick_interval)] = ticker_history exchange._klines[("UNITTEST/BTC", tick_interval)] = ticker_history
dp = DataProvider(default_conf, exchange) dp = DataProvider(default_conf, exchange)
assert dp.runmode == RunMode.DRY_RUN assert dp.runmode == RunMode.DRY_RUN
assert ticker_history.equals(dp.ohlcv('UNITTEST/BTC', tick_interval)) assert ticker_history.equals(dp.ohlcv("UNITTEST/BTC", tick_interval))
assert isinstance(dp.ohlcv('UNITTEST/BTC', tick_interval), DataFrame) assert isinstance(dp.ohlcv("UNITTEST/BTC", tick_interval), DataFrame)
assert dp.ohlcv('UNITTEST/BTC', tick_interval) is not ticker_history assert dp.ohlcv("UNITTEST/BTC", tick_interval) is not ticker_history
assert dp.ohlcv('UNITTEST/BTC', tick_interval, copy=False) is ticker_history assert dp.ohlcv("UNITTEST/BTC", tick_interval, copy=False) is ticker_history
assert not dp.ohlcv('UNITTEST/BTC', tick_interval).empty assert not dp.ohlcv("UNITTEST/BTC", tick_interval).empty
assert dp.ohlcv('NONESENSE/AAA', tick_interval).empty assert dp.ohlcv("NONESENSE/AAA", tick_interval).empty
# Test with and without parameter # Test with and without parameter
assert dp.ohlcv('UNITTEST/BTC', tick_interval).equals(dp.ohlcv('UNITTEST/BTC')) assert dp.ohlcv("UNITTEST/BTC", tick_interval).equals(dp.ohlcv("UNITTEST/BTC"))
default_conf['runmode'] = RunMode.LIVE default_conf["runmode"] = RunMode.LIVE
dp = DataProvider(default_conf, exchange) dp = DataProvider(default_conf, exchange)
assert dp.runmode == RunMode.LIVE assert dp.runmode == RunMode.LIVE
assert isinstance(dp.ohlcv('UNITTEST/BTC', tick_interval), DataFrame) assert isinstance(dp.ohlcv("UNITTEST/BTC", tick_interval), DataFrame)
default_conf['runmode'] = RunMode.BACKTEST default_conf["runmode"] = RunMode.BACKTEST
dp = DataProvider(default_conf, exchange) dp = DataProvider(default_conf, exchange)
assert dp.runmode == RunMode.BACKTEST assert dp.runmode == RunMode.BACKTEST
assert dp.ohlcv('UNITTEST/BTC', tick_interval).empty assert dp.ohlcv("UNITTEST/BTC", tick_interval).empty
def test_historic_ohlcv(mocker, default_conf, ticker_history): def test_historic_ohlcv(mocker, default_conf, ticker_history):
historymock = MagicMock(return_value=ticker_history) historymock = MagicMock(return_value=ticker_history)
mocker.patch('freqtrade.data.dataprovider.load_pair_history', historymock) mocker.patch("freqtrade.data.dataprovider.load_pair_history", historymock)
# exchange = get_patched_exchange(mocker, default_conf) # exchange = get_patched_exchange(mocker, default_conf)
dp = DataProvider(default_conf, None) dp = DataProvider(default_conf, None)
data = dp.historic_ohlcv('UNITTEST/BTC', "5m") data = dp.historic_ohlcv("UNITTEST/BTC", "5m")
assert isinstance(data, DataFrame) assert isinstance(data, DataFrame)
assert historymock.call_count == 1 assert historymock.call_count == 1
assert historymock.call_args_list[0][1]['datadir'] is None assert historymock.call_args_list[0][1]["datadir"] is None
assert historymock.call_args_list[0][1]['refresh_pairs'] is False assert historymock.call_args_list[0][1]["refresh_pairs"] is False
assert historymock.call_args_list[0][1]['ticker_interval'] == '5m' assert historymock.call_args_list[0][1]["ticker_interval"] == "5m"
def test_available_pairs(mocker, default_conf, ticker_history): def test_available_pairs(mocker, default_conf, ticker_history):
exchange = get_patched_exchange(mocker, default_conf) exchange = get_patched_exchange(mocker, default_conf)
tick_interval = default_conf['ticker_interval'] tick_interval = default_conf["ticker_interval"]
exchange._klines[('XRP/BTC', tick_interval)] = ticker_history exchange._klines[("XRP/BTC", tick_interval)] = ticker_history
exchange._klines[('UNITTEST/BTC', tick_interval)] = ticker_history exchange._klines[("UNITTEST/BTC", tick_interval)] = ticker_history
dp = DataProvider(default_conf, exchange) dp = DataProvider(default_conf, exchange)
assert len(dp.available_pairs) == 2 assert len(dp.available_pairs) == 2
assert dp.available_pairs == [('XRP/BTC', tick_interval), ('UNITTEST/BTC', tick_interval)] assert dp.available_pairs == [
("XRP/BTC", tick_interval),
("UNITTEST/BTC", tick_interval),
]
def test_refresh(mocker, default_conf, ticker_history): def test_refresh(mocker, default_conf, ticker_history):
refresh_mock = MagicMock() refresh_mock = MagicMock()
mocker.patch('freqtrade.exchange.Exchange.refresh_latest_ohlcv', refresh_mock) mocker.patch("freqtrade.exchange.Exchange.refresh_latest_ohlcv", refresh_mock)
exchange = get_patched_exchange(mocker, default_conf, id='binance') exchange = get_patched_exchange(mocker, default_conf, id="binance")
tick_interval = default_conf['ticker_interval'] tick_interval = default_conf["ticker_interval"]
pairs = [('XRP/BTC', tick_interval), pairs = [("XRP/BTC", tick_interval), ("UNITTEST/BTC", tick_interval)]
('UNITTEST/BTC', tick_interval),
]
pairs_non_trad = [('ETH/USDT', tick_interval), pairs_non_trad = [("ETH/USDT", tick_interval), ("BTC/TUSD", "1h")]
('BTC/TUSD', "1h"),
]
dp = DataProvider(default_conf, exchange) dp = DataProvider(default_conf, exchange)
dp.refresh(pairs) dp.refresh(pairs)