File size: 2,304 Bytes
079c32c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
import pytest
from ding.utils.autolog import TimeRangedData, NaturalTime, TickTime
@pytest.mark.unittest
class TestAutologRangedData:
def test_expire(self):
data = TimeRangedData(NaturalTime(), expire=5)
assert data.expire == 5
with pytest.raises(ValueError):
TimeRangedData(NaturalTime(), expire=-1)
with pytest.raises(TypeError):
TimeRangedData(NaturalTime(), expire='5')
def test_bool(self):
data = TimeRangedData(TickTime(), expire=5)
assert not data
data.append(233)
assert data
data.time.step()
data.extend([2, 3, 5, 7])
assert data
data.time.step(4)
assert data
data.time.step(1)
assert data
data.time.step(1)
assert data
data.time.step(1)
assert data
data.time.step(10)
assert data
def test_current(self):
data = TimeRangedData(TickTime(), expire=5)
with pytest.raises(ValueError):
_ = data.current()
data.append(233)
assert data.current() == 233
data.time.step()
data.extend([2, 3, 5, 7])
assert data.current() == 7
data.time.step(4)
assert data.current() == 7
data.time.step(1)
assert data.current() == 7
data.time.step(1)
assert data.current() == 7
data.time.step(1)
assert data.current() == 7
data.time.step(10)
assert data.current() == 7
def test_history(self):
data = TimeRangedData(TickTime(), expire=5)
assert data.history() == []
data.append(233)
assert data.history() == [(0, 233)]
data.time.step()
data.extend([2, 3, 5, 7])
assert data.history() == [(0, 233), (1, 2), (1, 3), (1, 5), (1, 7)]
data.time.step(4)
assert data.history() == [(0, 233), (1, 2), (1, 3), (1, 5), (1, 7), (5, 7)]
data.time.step(1)
assert data.history() == [(1, 233), (1, 2), (1, 3), (1, 5), (1, 7), (6, 7)]
data.time.step(1)
assert data.history() == [(2, 7), (7, 7)]
data.time.step(1)
assert data.history() == [(3, 7), (8, 7)]
data.time.step(10)
assert data.history() == [(13, 7), (18, 7)]
|