Python flatten Examples

Example #1

File: test_utils.py Project: 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]

Example #2

File: test_utils.py Project: kroger/aristoxenus

def test_flatten_two_levels():
    assert list(utils.flatten([[1, 2, [3, 3]], [4, 5]])) == [1, 2, 3, 3, 4, 5]

Example #3

File: test_utils.py Project: kroger/aristoxenus

def test_flatten_two_levels_and_empty_list():
    assert list(utils.flatten([[1, 2, []], [4, 5]])) == [1, 2, 4, 5]

Example #4

File: test_utils.py Project: kroger/aristoxenus

def test_flatten_one_level():
    assert list(utils.flatten([[1, 2], [4, 5]])) == [1, 2, 4, 5]