34 lines
720 B
Python
34 lines
720 B
Python
import pytest
|
|
|
|
from bonobo.util import sortedlist, ensure_tuple
|
|
from bonobo.util.collections import tuplize, cast
|
|
|
|
|
|
def test_sortedlist():
|
|
l = sortedlist()
|
|
l.insort(2)
|
|
l.insort(1)
|
|
l.insort(3)
|
|
l.insort(2)
|
|
assert l == [1, 2, 2, 3]
|
|
|
|
|
|
def test_ensure_tuple():
|
|
assert ensure_tuple('a') == ('a', )
|
|
assert ensure_tuple(('a', )) == ('a', )
|
|
assert ensure_tuple(()) is ()
|
|
|
|
|
|
@pytest.mark.parametrize('tuplize', [tuplize, cast(tuple)])
|
|
def test_tuplize(tuplize):
|
|
tuplized_lambda = tuplize(lambda: [1, 2, 3])
|
|
assert tuplized_lambda() == (1, 2, 3)
|
|
|
|
@tuplize
|
|
def some_generator():
|
|
yield 'c'
|
|
yield 'b'
|
|
yield 'a'
|
|
|
|
assert some_generator() == ('c', 'b', 'a')
|