84 lines
3.2 KiB
Python
84 lines
3.2 KiB
Python
# pragma pylint: disable=missing-docstring,C0103
|
|
|
|
import datetime
|
|
from unittest.mock import MagicMock
|
|
|
|
from freqtrade.data.converter import parse_ticker_dataframe
|
|
from freqtrade.misc import (common_datearray, datesarray_to_datetimearray,
|
|
file_dump_json, file_load_json, format_ms_time, shorten_date)
|
|
from freqtrade.data.history import load_tickerdata_file, pair_data_filename
|
|
from freqtrade.strategy.default_strategy import DefaultStrategy
|
|
|
|
|
|
def test_shorten_date() -> None:
|
|
str_data = '1 day, 2 hours, 3 minutes, 4 seconds ago'
|
|
str_shorten_data = '1 d, 2 h, 3 min, 4 sec ago'
|
|
assert shorten_date(str_data) == str_shorten_data
|
|
|
|
|
|
def test_datesarray_to_datetimearray(ticker_history_list):
|
|
dataframes = parse_ticker_dataframe(ticker_history_list, "5m", fill_missing=True)
|
|
dates = datesarray_to_datetimearray(dataframes['date'])
|
|
|
|
assert isinstance(dates[0], datetime.datetime)
|
|
assert dates[0].year == 2017
|
|
assert dates[0].month == 11
|
|
assert dates[0].day == 26
|
|
assert dates[0].hour == 8
|
|
assert dates[0].minute == 50
|
|
|
|
date_len = len(dates)
|
|
assert date_len == 2
|
|
|
|
|
|
def test_common_datearray(default_conf) -> None:
|
|
strategy = DefaultStrategy(default_conf)
|
|
tick = load_tickerdata_file(None, 'UNITTEST/BTC', '1m')
|
|
tickerlist = {'UNITTEST/BTC': parse_ticker_dataframe(tick, "1m", fill_missing=True)}
|
|
dataframes = strategy.tickerdata_to_dataframe(tickerlist)
|
|
|
|
dates = common_datearray(dataframes)
|
|
|
|
assert dates.size == dataframes['UNITTEST/BTC']['date'].size
|
|
assert dates[0] == dataframes['UNITTEST/BTC']['date'][0]
|
|
assert dates[-1] == dataframes['UNITTEST/BTC']['date'].iloc[-1]
|
|
|
|
|
|
def test_file_dump_json(mocker) -> None:
|
|
file_open = mocker.patch('freqtrade.misc.open', MagicMock())
|
|
json_dump = mocker.patch('rapidjson.dump', MagicMock())
|
|
file_dump_json('somefile', [1, 2, 3])
|
|
assert file_open.call_count == 1
|
|
assert json_dump.call_count == 1
|
|
file_open = mocker.patch('freqtrade.misc.gzip.open', MagicMock())
|
|
json_dump = mocker.patch('rapidjson.dump', MagicMock())
|
|
file_dump_json('somefile', [1, 2, 3], True)
|
|
assert file_open.call_count == 1
|
|
assert json_dump.call_count == 1
|
|
|
|
|
|
def test_file_load_json(mocker) -> None:
|
|
|
|
# 7m .json does not exist
|
|
ret = file_load_json(pair_data_filename(None, 'UNITTEST/BTC', '7m'))
|
|
assert not ret
|
|
# 1m json exists (but no .gz exists)
|
|
ret = file_load_json(pair_data_filename(None, 'UNITTEST/BTC', '1m'))
|
|
assert ret
|
|
# 8 .json is empty and will fail if it's loaded. .json.gz is a copy of 1.json
|
|
ret = file_load_json(pair_data_filename(None, 'UNITTEST/BTC', '8m'))
|
|
assert ret
|
|
|
|
|
|
def test_format_ms_time() -> None:
|
|
# Date 2018-04-10 18:02:01
|
|
date_in_epoch_ms = 1523383321000
|
|
date = format_ms_time(date_in_epoch_ms)
|
|
assert type(date) is str
|
|
res = datetime.datetime(2018, 4, 10, 18, 2, 1, tzinfo=datetime.timezone.utc)
|
|
assert date == res.astimezone(None).strftime('%Y-%m-%dT%H:%M:%S')
|
|
res = datetime.datetime(2017, 12, 13, 8, 2, 1, tzinfo=datetime.timezone.utc)
|
|
# Date 2017-12-13 08:02:01
|
|
date_in_epoch_ms = 1513152121000
|
|
assert format_ms_time(date_in_epoch_ms) == res.astimezone(None).strftime('%Y-%m-%dT%H:%M:%S')
|