mirror of
https://github.com/Textualize/textual.git
synced 2025-10-17 02:38:12 +03:00
66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
from textual._loop import loop_first, loop_first_last, loop_from_index, loop_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")
|
|
|
|
|
|
def test_loop_from_index():
|
|
assert list(loop_from_index("abcdefghij", 3)) == [
|
|
(4, "e"),
|
|
(5, "f"),
|
|
(6, "g"),
|
|
(7, "h"),
|
|
(8, "i"),
|
|
(9, "j"),
|
|
(0, "a"),
|
|
(1, "b"),
|
|
(2, "c"),
|
|
(3, "d"),
|
|
]
|
|
|
|
assert list(loop_from_index("abcdefghij", 3, direction=-1)) == [
|
|
(2, "c"),
|
|
(1, "b"),
|
|
(0, "a"),
|
|
(9, "j"),
|
|
(8, "i"),
|
|
(7, "h"),
|
|
(6, "g"),
|
|
(5, "f"),
|
|
(4, "e"),
|
|
(3, "d"),
|
|
]
|
|
|
|
assert list(loop_from_index("abcdefghij", 3, wrap=False)) == [
|
|
(4, "e"),
|
|
(5, "f"),
|
|
(6, "g"),
|
|
(7, "h"),
|
|
(8, "i"),
|
|
(9, "j"),
|
|
]
|