1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
from wn._util import flatten, unique_list
def test_flatten():
assert flatten([]) == []
assert flatten([[]]) == []
assert flatten([[], []]) == []
assert flatten([[[], []], [[], []]]) == [[], [], [], []]
assert flatten([[1]]) == [1]
assert flatten([[1, 2], [3, 4]]) == [1, 2, 3, 4]
assert flatten(["AB", "CD"]) == ["A", "B", "C", "D"]
def test_unique_list():
assert unique_list([]) == []
assert unique_list([1]) == [1]
assert unique_list([1, 1, 1, 1, 1]) == [1]
assert unique_list([1, 1, 2, 2, 1]) == [1, 2]
assert unique_list([2, 1, 2, 2, 1]) == [2, 1]
assert unique_list("A") == ["A"]
assert unique_list("AAA") == ["A"]
assert unique_list("ABABA") == ["A", "B"]
assert unique_list([(1, 2), (1, 2), (2, 3)]) == [(1, 2), (2, 3)]
|
