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

Reformat tests to be easier readable

Browse Source
This commit is contained in:
Matthias 2019-08-12 16:17:06 +02:00
parent dd0ba183f8
commit 1ce63b5b42
2 changed files with 40 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
freqtrade/exchange/exchange.py
View File

@ -793,9 +793,12 @@ def timeframe_to_msecs(ticker_interval: str) -> int:
return ccxt.Exchange.parse_timeframe(ticker_interval) * 1000 return ccxt.Exchange.parse_timeframe(ticker_interval) * 1000
def timeframe_to_prev_date(timeframe: str, date: datetime = None): def timeframe_to_prev_date(timeframe: str, date: datetime = None) -> datetime:
""" """
Use Timeframe and determine last possible candle. Use Timeframe and determine last possible candle.
:param timeframe: timeframe in string format (e.g. "5m")
:param date: date to use. Defaults to utcnow()
:returns: date of previous candle (with utc timezone)
""" """
if not date: if not date:
date = datetime.utcnow() date = datetime.utcnow()
@ -807,9 +810,12 @@ def timeframe_to_prev_date(timeframe: str, date: datetime = None):
return datetime.fromtimestamp(new_timestamp, tz=timezone.utc) return datetime.fromtimestamp(new_timestamp, tz=timezone.utc)
def timeframe_to_next_date(timeframe: str, date: datetime = None): def timeframe_to_next_date(timeframe: str, date: datetime = None) -> datetime:
""" """
Use Timeframe and determine next candle. Use Timeframe and determine next candle.
:param timeframe: timeframe in string format (e.g. "5m")
:param date: date to use. Defaults to utcnow()
:returns: date of next candle (with utc timezone)
""" """
if not date: if not date:
date = datetime.utcnow() date = datetime.utcnow()

46
freqtrade/tests/exchange/test_exchange.py
View File

@ -1571,45 +1571,41 @@ def test_timeframe_to_prev_date():
# 2019-08-12 13:22:08 # 2019-08-12 13:22:08
date = datetime.fromtimestamp(1565616128, tz=timezone.utc) date = datetime.fromtimestamp(1565616128, tz=timezone.utc)
tf_list = [
# 5m -> 2019-08-12 13:20:00 # 5m -> 2019-08-12 13:20:00
assert timeframe_to_prev_date("5m", date) == datetime( ("5m", datetime(2019, 8, 12, 13, 20, 0, tzinfo=timezone.utc)),
2019, 8, 12, 13, 20, 0, tzinfo=timezone.utc)
# 10m -> 2019-08-12 13:20:00 # 10m -> 2019-08-12 13:20:00
assert timeframe_to_prev_date("10m", date) == datetime( ("10m", datetime(2019, 8, 12, 13, 20, 0, tzinfo=timezone.utc)),
2019, 8, 12, 13, 20, 0, tzinfo=timezone.utc)
# 1h -> 2019-08-12 13:00:00 # 1h -> 2019-08-12 13:00:00
assert timeframe_to_prev_date("1h", date) == datetime( ("1h", datetime(2019, 8, 12, 13, 00, 0, tzinfo=timezone.utc)),
2019, 8, 12, 13, 00, 0, tzinfo=timezone.utc)
# 2h -> 2019-08-12 12:00:00 # 2h -> 2019-08-12 12:00:00
assert timeframe_to_prev_date("2h", date) == datetime( ("2h", datetime(2019, 8, 12, 12, 00, 0, tzinfo=timezone.utc)),
2019, 8, 12, 12, 00, 0, tzinfo=timezone.utc)
# 4h -> 2019-08-12 12:00:00 # 4h -> 2019-08-12 12:00:00
assert timeframe_to_prev_date("4h", date) == datetime( ("4h", datetime(2019, 8, 12, 12, 00, 0, tzinfo=timezone.utc)),
2019, 8, 12, 12, 00, 0, tzinfo=timezone.utc)
# 1d -> 2019-08-12 00:00:00 # 1d -> 2019-08-12 00:00:00
assert timeframe_to_prev_date("1d", date) == datetime( ("1d", datetime(2019, 8, 12, 00, 00, 0, tzinfo=timezone.utc)),
2019, 8, 12, 0, 0, 0, tzinfo=timezone.utc) ]
for interval, result in tf_list:
assert timeframe_to_prev_date(interval, date) == result
def test_timeframe_to_next_date(): def test_timeframe_to_next_date():
# 2019-08-12 13:22:08 # 2019-08-12 13:22:08
date = datetime.fromtimestamp(1565616128, tz=timezone.utc) date = datetime.fromtimestamp(1565616128, tz=timezone.utc)
tf_list = [
# 5m -> 2019-08-12 13:25:00 # 5m -> 2019-08-12 13:25:00
assert timeframe_to_next_date("5m", date) == datetime( ("5m", datetime(2019, 8, 12, 13, 25, 0, tzinfo=timezone.utc)),
2019, 8, 12, 13, 25, 0, tzinfo=timezone.utc)
# 10m -> 2019-08-12 13:30:00 # 10m -> 2019-08-12 13:30:00
assert timeframe_to_next_date("10m", date) == datetime( ("10m", datetime(2019, 8, 12, 13, 30, 0, tzinfo=timezone.utc)),
2019, 8, 12, 13, 30, 0, tzinfo=timezone.utc)
# 1h -> 2019-08-12 14:00:00 # 1h -> 2019-08-12 14:00:00
assert timeframe_to_next_date("1h", date) == datetime( ("1h", datetime(2019, 8, 12, 14, 00, 0, tzinfo=timezone.utc)),
2019, 8, 12, 14, 00, 0, tzinfo=timezone.utc)
# 2h -> 2019-08-12 14:00:00 # 2h -> 2019-08-12 14:00:00
assert timeframe_to_next_date("2h", date) == datetime( ("2h", datetime(2019, 8, 12, 14, 00, 0, tzinfo=timezone.utc)),
2019, 8, 12, 14, 00, 0, tzinfo=timezone.utc)
# 4h -> 2019-08-12 14:00:00 # 4h -> 2019-08-12 14:00:00
assert timeframe_to_next_date("4h", date) == datetime( ("4h", datetime(2019, 8, 12, 16, 00, 0, tzinfo=timezone.utc)),
2019, 8, 12, 16, 00, 0, tzinfo=timezone.utc)
# 1d -> 2019-08-13 00:00:00 # 1d -> 2019-08-13 00:00:00
assert timeframe_to_next_date("1d", date) == datetime( ("1d", datetime(2019, 8, 13, 0, 0, 0, tzinfo=timezone.utc)),
2019, 8, 13, 0, 0, 0, tzinfo=timezone.utc) ]
for interval, result in tf_list:
assert timeframe_to_next_date(interval, date) == result