Reformat tests to be easier readable

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

View File

@ -793,9 +793,12 @@ def timeframe_to_msecs(ticker_interval: str) -> int:
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.
: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:
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)
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.
: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:
date = datetime.utcnow()

View File

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