mirror of
https://github.com/pypa/pipenv.git
synced 2026-05-07 00:39:57 -04:00
Matt Davis
56 lines
1.1 KiB
Python
56 lines
1.1 KiB
Python
import pytest
|
|
|
|
from pipenv.utils.funktools import _is_iterable, dedup, unnest
|
|
|
|
|
|
def test_unnest():
|
|
nested_iterable = (
|
|
1234,
|
|
(3456, 4398345, (234234)),
|
|
(2396, (928379, 29384, (293759, 2347, (2098, 7987, 27599)))),
|
|
)
|
|
assert list(unnest(nested_iterable)) == [
|
|
1234,
|
|
3456,
|
|
4398345,
|
|
234234,
|
|
2396,
|
|
928379,
|
|
29384,
|
|
293759,
|
|
2347,
|
|
2098,
|
|
7987,
|
|
27599,
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"iterable, result",
|
|
[
|
|
[["abc", "def"], True],
|
|
[("abc", "def"), True],
|
|
["abcdef", True],
|
|
[None, False],
|
|
[1234, False],
|
|
],
|
|
)
|
|
def test_is_iterable(iterable, result):
|
|
assert _is_iterable(iterable) is result
|
|
|
|
|
|
def test_unnest_none():
|
|
assert list(unnest(None)) == [None]
|
|
|
|
|
|
def test_dedup():
|
|
dup_strings = ["abcde", "fghij", "klmno", "pqrst", "abcde", "klmno"]
|
|
assert list(dedup(dup_strings)) == [
|
|
"abcde",
|
|
"fghij",
|
|
"klmno",
|
|
"pqrst",
|
|
]
|
|
dup_ints = (12345, 56789, 12345, 54321, 98765, 54321)
|
|
assert list(dedup(dup_ints)) == [12345, 56789, 54321, 98765]
|