Generalize tuplize decorator to cast(...) decorator.

This commit is contained in:
Romain Dorgueil
2017-12-27 14:21:50 +01:00
parent f64e6e3c76
commit 464d08a4a2
5 changed files with 40 additions and 29 deletions

View File

@ -1,5 +1,7 @@
import pytest
from bonobo.util import sortedlist, ensure_tuple
from bonobo.util.collections import tuplize
from bonobo.util.collections import tuplize, cast
def test_sortedlist():
@ -12,12 +14,13 @@ def test_sortedlist():
def test_ensure_tuple():
assert ensure_tuple('a') == ('a', )
assert ensure_tuple(('a', )) == ('a', )
assert ensure_tuple('a') == ('a',)
assert ensure_tuple(('a',)) == ('a',)
assert ensure_tuple(()) is ()
def test_tuplize():
@pytest.mark.parametrize('tuplize', [tuplize, cast(tuple)])
def test_tuplize(tuplize):
tuplized_lambda = tuplize(lambda: [1, 2, 3])
assert tuplized_lambda() == (1, 2, 3)