# py-scopes: tests.test_util """Tests for the scopes.util package""" import pytest from scopes.util import keywords as k, flatten k('dummy', 'other', 'cd', 'new') 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'] 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]