aniani/toot
0
0
Fork 0
mirror of https://github.com/ihabunek/toot.git synced 2025-10-21 19:44:16 -04:00
Code Activity

Add batched helper

Browse Source
This commit is contained in:
Ivan Habunek
2025-01-12 12:00:59 +01:00
parent c98af14a84
commit 38dfd747f6
2 changed files with 31 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_utils.py
View File

@@ -7,7 +7,7 @@ from toot.wcstring import wc_wrap, trunc, pad, fit_text
from toot.tui.utils import LRUCache
from PIL import Image
from collections import namedtuple
from toot.utils import urlencode_url
from toot.utils import batched, urlencode_url
def test_pad():
@@ -319,3 +319,15 @@ def test_urlencode_url():
assert urlencode_url("https://www.example.com") == "https://www.example.com"
assert urlencode_url("https://www.example.com/url%20with%20spaces") == "https://www.example.com/url%20with%20spaces"
def test_batched():
assert list(batched("", 2)) == []
assert list(batched("a", 2)) == [["a"]]
assert list(batched("ab", 2)) == [["a", "b"]]
assert list(batched("abc", 2)) == [["a", "b"], ["c"]]
assert list(batched("abcd", 2)) == [["a", "b"], ["c", "d"]]
assert list(batched("abcde", 2)) == [["a", "b"], ["c", "d"], ["e"]]
assert list(batched("abcdef", 2)) == [["a", "b"], ["c", "d"], ["e", "f"]]
with pytest.raises(ValueError):
list(batched("foo", 0))