22 lines
587 B
Python
22 lines
587 B
Python
from layercake.funcs import omit, pick
|
|
|
|
|
|
def test_omit():
|
|
values = {'indigo': '#4b0082', 'navy': '#000080'}
|
|
assert omit(['indigo'], values) == {'navy': '#000080'}
|
|
assert omit(['test'], values) == values
|
|
|
|
|
|
def test_pick():
|
|
values = {'indigo': '#4b0082', 'navy': '#000080'}
|
|
assert pick(['navy'], values) == {'navy': '#000080'}
|
|
assert pick(['test'], values) == {}
|
|
|
|
|
|
def test_pick_default_val():
|
|
values = {'name': 'test'}
|
|
assert pick(['name', 'surname'], values, exclude_none=False, default=False) == {
|
|
'name': 'test',
|
|
'surname': False,
|
|
}
|