mirror of
https://github.com/freqtrade/freqtrade.git
synced 2024-11-10 10:21:59 +00:00
Add test for measure_time
This commit is contained in:
parent
c6a50b3c06
commit
5eeb96fa96
34
tests/utils/test_measure_time.py
Normal file
34
tests/utils/test_measure_time.py
Normal file
|
@ -0,0 +1,34 @@
|
|||
from unittest.mock import MagicMock
|
||||
|
||||
import time_machine
|
||||
|
||||
from freqtrade.util import MeasureTime
|
||||
|
||||
|
||||
def test_measure_time():
|
||||
|
||||
callback = MagicMock()
|
||||
with time_machine.travel("2021-09-01 05:00:00 +00:00", tick=False) as t:
|
||||
|
||||
measure = MeasureTime(callback, 5, ttl=60)
|
||||
with measure:
|
||||
pass
|
||||
|
||||
assert callback.call_count == 0
|
||||
|
||||
with measure:
|
||||
t.shift(10)
|
||||
|
||||
assert callback.call_count == 1
|
||||
callback.reset_mock()
|
||||
with measure:
|
||||
t.shift(10)
|
||||
assert callback.call_count == 0
|
||||
|
||||
callback.reset_mock()
|
||||
# Shift past the ttl
|
||||
t.shift(45)
|
||||
|
||||
with measure:
|
||||
t.shift(10)
|
||||
assert callback.call_count == 1
|
Loading…
Reference in New Issue
Block a user