Python assertEquals生成器

时间:2014-08-18 13:56:32

标签: python

我正在测试包含对象的生成器对象。我想检查对象值属性,看它是否包含数字,如预期的那样。

生成器对象包含来自Test类的5个对象实例。所以我想做这样的事情:

伪代码: 生成器对象(列表)中的所有Test实例对象的value属性(test.value)等于[1,2,3,4,5]。

使用一行for循环或assertEquals方法中的某些内容是否可以实现?

1 个答案:

答案 0 :(得分:1)

因为它只有5个元素,所以只需使用列表理解。

assertEqual([test.value for test in test_instance_objects], [1, 2, 3, 4, 5])