How can you determine if a function return value is a generator?

Question

the yield return a collections.Iterable but many types are also iterable

def aaaa():
    yield 1
    yield 2
    yield 3 

d = aaaa()

print(d,type(d),isinstance(d, collections.Iterable))

the print type

<generator object aaaa at 0x0000000002626B88> <class 'generator'> True

i didn't find the class 'generator' or i can't do typecheck.

Answer 1

If you want to check if the iterable is a generator object:

import types
isinstance(aaaa(), types.GeneratorType) # ==> True

If you want to check if the function contains yield statements (ie the function is a generator):

import inspect
inspect.isgeneratorfunction(aaaa) # ==> True

Answer 2

Have a look at the inspect module, and in particular inspect.isgeneratorfunction .

That said, the more interesting question is whether it's a list or other iterable. To answer that question, you can just check if __getitem__ is defined on it.