Python NestedIterator Examples

Example #1

File: test.py Project: misaka-10032/leetcode

def test_1():
    it = NestedIterator([NestedInteger([NestedInteger(1), NestedInteger(1)]),
                         NestedInteger(2),
                         NestedInteger([NestedInteger(1),
                                        NestedInteger([NestedInteger(4),
                                                       NestedInteger(6)])]),
                         NestedInteger(9)])
    res = []
    while it.hasNext():
        res.append(it.next())
    assert res == [1, 1, 2, 1, 4, 6, 9]

Example #2

File: test.py Project: misaka-10032/leetcode

def test_0():
    it = NestedIterator([])
    assert not it.hasNext()

    it = NestedIterator([NestedInteger(1)])
    assert it.hasNext()
    assert it.next() == 1
    assert not it.hasNext()

    it = NestedIterator([NestedInteger([NestedInteger([NestedInteger(1)])])])
    assert it.hasNext()
    assert it.next() == 1
    assert not it.hasNext()