Unit testing a function that returns a generator object

Question

The title pretty much sums it up: I tried to use assertEqual to test a function that returns a generator object, but that results in:

AssertionError: gene

Answer 1

Use next() on the generator object.

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