Improve dt_now_ts helper

This commit is contained in:
Matthias
2023-05-14 10:45:07 +02:00
parent aa949153eb
commit 000f72942a
3 changed files with 14 additions and 6 deletions

View File

@@ -3,19 +3,24 @@ from datetime import datetime, timedelta, timezone
import pytest
import time_machine
from freqtrade.util import dt_from_ts, dt_now, dt_now_ts
from freqtrade.util import dt_from_ts, dt_now, dt_ts
def test_dt_now():
with time_machine.travel("2021-09-01 05:01:00 +00:00", tick=False) as t:
now = datetime.now(timezone.utc)
assert dt_now() == now
assert dt_now_ts() == int(now.timestamp() * 1000)
assert dt_ts() == int(now.timestamp() * 1000)
assert dt_ts(now) == int(now.timestamp() * 1000)
t.shift(timedelta(hours=5))
assert dt_now() >= now
assert dt_now() == datetime.now(timezone.utc)
assert dt_now_ts() == int(dt_now().timestamp() * 1000)
assert dt_ts() == int(dt_now().timestamp() * 1000)
# Test with different time than now
assert dt_ts(now) == int(now.timestamp() * 1000)
@pytest.mark.parametrize('as_ms', [True, False])