Python flatten Examples

Example #1

def test_unit_remove_multi_level():
    a = Tuple[Tuple[Uint[1], Unit, Uint[2]], Unit, Uint[3]]
    assert flatten(a, lvl=2) == Tuple[Uint[1], Uint[2], Uint[3]]

Example #2

def test_unit_remove_single_level():
    a = Tuple[Tuple[Uint[1], Unit, Uint[2]], Unit, Uint[3]]
    assert flatten(a) == Tuple[Uint[1], Unit, Uint[2], Uint[3]]

Example #3

def test_vanish():
    a = Tuple[Tuple[Unit, Tuple[Unit]]]
    assert flatten(a, lvl=2) == Unit

Example #4

def test_complex():
    a = Tuple[Tuple[Uint[1], Unit, Tuple[Tuple[Uint[2]]]], Tuple[Unit],
              Uint[3]]
    assert flatten(a, lvl=2) == Tuple[Uint[1], Tuple[Uint[2]], Uint[3]]

Example #5

def test_resolve_multi_level():
    a = Tuple[Tuple[Uint[1]]]
    assert flatten(a, lvl=2) == Uint[1]

Example #6

def test_resolve_single_level():
    a = Tuple[Tuple[Tuple[Uint[1]]]]
    assert flatten(a) == Tuple[Uint[1]]