コード例 #1
0
ファイル: test_utils.py プロジェクト: kroger/aristoxenus 
def test_flatten_three_levels():
    big_list = [[1, 2, [3, [4, 5]]], [6, [[7, 8]], 9]]
    assert list(utils.flatten(big_list)) == [1, 2, 3, 4, 5, 6, 7, 8, 9]
コード例 #2
0
ファイル: test_utils.py プロジェクト: kroger/aristoxenus 
def test_flatten_two_levels():
    assert list(utils.flatten([[1, 2, [3, 3]], [4, 5]])) == [1, 2, 3, 3, 4, 5]
コード例 #3
0
ファイル: test_utils.py プロジェクト: kroger/aristoxenus 
def test_flatten_two_levels_and_empty_list():
    assert list(utils.flatten([[1, 2, []], [4, 5]])) == [1, 2, 4, 5]
コード例 #4
0
ファイル: test_utils.py プロジェクト: kroger/aristoxenus 
def test_flatten_one_level():
    assert list(utils.flatten([[1, 2], [4, 5]])) == [1, 2, 4, 5]