单元测试返回生成器对象的函数

Question

标题几乎总结了一下：我试图使用assertEqual来测试一个返回生成器对象的函数，但结果是：

  

AssertionError：generator object genexpr＆gt;在0x2e70c80＆gt; ！=（1,2，...）

除了迭代函数并为每个输出运行assertEqual之外，还有一种聪明的方法吗？

Answer 1

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

如果它是无限生成器，或者您只是想出于某种原因查看第一个n结果，可以将其与itertools.islice

结合使用

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

Answer 2

在生成器对象上使用next()。

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