Unidad de prueba de una función que devuelve un objeto generador de

Question

El título prácticamente lo resume todo: He intentado utilizar assertEqual para probar una función que devuelve un objeto generador, pero que se traduce en:

AssertionError: generator object genexpr> at 0x2e70c80> != (1,2,...)

¿Hay una forma inteligente de hacer esto aparte de iterar sobre la función y ejecutar assertEqual para cada salida individual?

Answer 1

assertEqual(tuple(generator_object), (1, 2, ...)) 

si se trata de un generador infinito, o si sólo desea ver los primeros resultados n por alguna razón, puede combint esto con itertools.islice

assertEqual(tuple(islice(generator_object, n)), (1, 2, ...)) 

Answer 2

Uso next() en el objeto generador.

assertEqual(next(generator_object), (1, 2, ...))