mirror of
https://github.com/Textualize/textual.git
synced 2025-10-17 02:38:12 +03:00
57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
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
|