Haskell-style infinite "list" testing with Python.

infinite.py

"""
Haskell-style infinite "list" testing.
"""


def infinite_range(step):
    i = 0
    while True:
        yield i
        i += step


def take(count, iterator):
    a = []
    for x in xrange(count):
        try:
            a.append(iterator.next())
        except StopIteration:
            pass
    return a


def main():
    print take(5, infinite_range(1))

    print take(10, infinite_range(5))

    million_items = take(10**6, infinite_range(1))
    print len(million_items)


if __name__ == '__main__':
    main()