functools.partial и генераторы

Question

Я пытаюсь сделать следующее:

import functools 

class TestClass(): 
    def method(self, n): 
     for i in xrange(n): 
      yield i 

# This works 
for x in TestClass().method(10): 
    print x 

# This gets a TypeError: functools.partial object not iterable 
for x in functools.partial(TestClass().method, 10): 
    print x 

Что плохого?

Answer 1

functools.partial создает object, который ведет себя как новая функция, которая имитирует старую функцию с некоторыми замороженными аргументами. Таким образом, вы должны называть эту новую функцию для получения того же выхода:

for x in functools.partial(TestClass().method, 10)(): 
    print x