~homeworkprod/byceps

ref: ee44f550bd6c9f49e72424cd5826669dd1f57c1c byceps/tests/unit/util/iterables/test_partition.py -rw-r--r-- 594 bytes
ee44f550 — Jochen Kupperschmidt Avoid same German translation for different meanings of "cancel" 1 year, 8 months ago
                                                                                
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
"""
:Copyright: 2006-2021 Jochen Kupperschmidt
:License: Revised BSD (see `LICENSE` file for details)
"""

import pytest

from byceps.util.iterables import partition


@pytest.mark.parametrize(
    'iterable, predicate, expected',
    [
        (
            [],
            lambda x: x > 3,
            ([], []),
        ),
        (
            [0, 1, 2, 3, 4, 5, 6, 7],
            lambda x: x % 3 == 0,
            ([0, 3, 6], [1, 2, 4, 5, 7]),
        ),
    ],
)
def test_partition(iterable, predicate, expected):
    actual = partition(iterable, predicate)
    assert actual == expected