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
|
from textual.eta import ETA
def test_basics() -> None:
eta = ETA()
eta.add_sample(1.0, 1.0)
assert eta.first_sample == (0, 0)
assert eta.last_sample == (1.0, 1.0)
assert len(eta._samples) == 2
repr(eta)
def test_speed() -> None:
eta = ETA()
# One sample is not enough to determine speed
assert eta.speed is None
eta.add_sample(1.0, 0.5)
assert eta.speed == 0.5
# Check reset
eta.reset()
assert eta.speed is None
eta.add_sample(0.0, 0.0)
assert eta.speed is None
eta.add_sample(1.0, 0.5)
assert eta.speed == 0.5
# Check backwards
eta.add_sample(2.0, 0.0)
assert eta.speed is None
eta.add_sample(3.0, 1.0)
assert eta.speed == 1.0
def test_get_progress_at() -> None:
eta = ETA()
eta.add_sample(1, 2)
# Check interpolation works
assert eta._get_progress_at(0) == (0, 0)
assert eta._get_progress_at(0.1) == (0.1, 0.2)
assert eta._get_progress_at(0.5) == (0.5, 1.0)
def test_eta():
eta = ETA(estimation_period=2, extrapolate_period=5)
eta.add_sample(1, 0.1)
assert eta.speed == 0.1
assert eta.get_eta(1) == 9
assert eta.get_eta(2) == 8
assert eta.get_eta(3) == 7
assert eta.get_eta(4) == 6
assert eta.get_eta(5) == 5
# After 5 seconds (extrapolate_period), eta won't update
assert eta.get_eta(6) == 4
assert eta.get_eta(7) == 4
assert eta.get_eta(8) == 4
|
