为什么我的Python测试生成器根本不起作用?

时间:2009-07-09 02:30:01

标签: python testing generator yield

这是一个测试使用率的示例脚本......我做错了吗?它总是返回'1'......

#!/usr/bin/python

def testGen():
    for a in [1,2,3,4,5,6,7,8,9,10]:
         yield a

w = 0
while w < 10:
    print testGen().next()
        w += 1

1 个答案:

答案 0 :(得分:10)

每次都在创建一个新的生成器。您应该只调用testGen()一次,然后使用返回的对象。尝试:

w = 0
g = testGen()
while w < 10:
    print g.next()
    w += 1

当然,这是正常的,惯用的发电机用法:

for n in testGen():
    print n

请注意,这只会在循环开始时调用testGen()一次,而不是每次迭代调用一次。