This commit is contained in:
Will McGugan
2021-08-31 17:53:41 +01:00
parent 3f28a36028
commit 3dba37e550
5 changed files with 53 additions and 10 deletions

28
tests/test_loop.py Normal file
View File

@@ -0,0 +1,28 @@
from textual._loop import loop_first, loop_last, loop_first_last
def test_loop_first():
assert list(loop_first([])) == []
iterable = loop_first(["apples", "oranges", "pears", "lemons"])
assert next(iterable) == (True, "apples")
assert next(iterable) == (False, "oranges")
assert next(iterable) == (False, "pears")
assert next(iterable) == (False, "lemons")
def test_loop_last():
assert list(loop_last([])) == []
iterable = loop_last(["apples", "oranges", "pears", "lemons"])
assert next(iterable) == (False, "apples")
assert next(iterable) == (False, "oranges")
assert next(iterable) == (False, "pears")
assert next(iterable) == (True, "lemons")
def test_loop_first_last():
assert list(loop_first_last([])) == []
iterable = loop_first_last(["apples", "oranges", "pears", "lemons"])
assert next(iterable) == (True, False, "apples")
assert next(iterable) == (False, False, "oranges")
assert next(iterable) == (False, False, "pears")
assert next(iterable) == (False, True, "lemons")