21 lines
601 B
Python
21 lines
601 B
Python
# py-scopes: tests.test_util
|
|
|
|
"""Tests for the scopes.util package"""
|
|
|
|
import pytest
|
|
|
|
from scopes.util import keywords as k, keyword_dict, flatten
|
|
|
|
def test_keywords():
|
|
assert k.dummy is k.dummy
|
|
assert k.dummy is not k.other
|
|
assert getattr(k, "abcd"[-2:]) is k.cd
|
|
assert k('dummy') is k.dummy
|
|
assert k('new') is k('new')
|
|
assert keyword_dict(dict(one=1, two=2)) == {k.one: 1, k.two: 2}
|
|
|
|
def test_flatten():
|
|
lst = (1, (None, 2), (3, (4, 5)))
|
|
assert flatten(lst) == [1, None, 2, 3, 4, 5]
|
|
assert flatten(42, lst, (6, 7), ignore_none=True) == [42, 1, 2, 3, 4, 5, 6, 7]
|
|
|