Unità testare una funzione che restituisce un oggetto generatore

Question

Il titolo praticamente riassume: ho cercato di usare assertEqual per testare una funzione che restituisce un oggetto generatore, ma che si traduce in:

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

c'è un modo intelligente per farlo a parte l'iterazione sulla funzione e l'esecuzione di assertEqual per ogni singolo output?

Answer 1

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

se si tratta di un generatore di infinito, o semplicemente desidera guardare i primi risultati n per qualche motivo, è possibile combint questo con itertools.islice

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

Answer 2

Usa next() sull'oggetto generatore.

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