Тестирование генератора в Python

Question

Я запрограммировал очень простой генератор в Python 3.4:

class Gen: 
    def __init__(self, xml_lines, attribs): 
    #... 

    def _get_values(self, xml_line): 
    # get only specific "attribs" from a line in a "xml" 

    def values(self): 
    for line in self.xml_lines: 
     yield self._get_values(line) 

код работает, когда я использую для цикла, чтобы потреблять сгенерированные значения:

reader = Gen(...): 
for v in reader.values(): 
    print(v) 

Но я m теперь пытаюсь создать единичный тест, и для этого мне нужно будет получать каждое значение за раз (вне цикла). Я пытаюсь что-то вроде этого:

import unittest 
#... 
reader = Gen(...): 
v = reader.values() 
self.assertIsNone(v) 

Когда я пытаюсь это, я всегда получаю

AssertionError: <generator object next_values at 0x7fe2a5f9e3f0> is not None 

Так что, когда я называю значения себя, что-то вернуть (это указатель ?) вместо генерируемого значения.

Поскольку я рассматриваю, что использую базовый генераторный шаблон, мой вопрос немного шире, чем мой собственный код: каков правильный способ тестирования генератора Python?

Answer 1

Генератор предназначен для повторения. Вы можете использовать функцию next(), чтобы получить следующее значение генератора. Имейте в виду, что если ваш генератор исчерпал свои значения, он поднимет исключение StopIteration.

reader = Gen(...): 
values = reader.values() 
v = next(values) 
self.assertIsNone(v) 

Answer 2

мне нужно модульного тестирования генератора себя, поэтому я создал эту вспомогательную функцию:

def generator_tester(generator_iterator_to_test, expected_values): 
    range_index = 0 
    for actual in generator_iterator_to_test: 
     assert range_index + 1 <= len(expected_values), 'Too many values returned from range' 
     assert expected_values[range_index] == actual 
     range_index += 1 

    assert range_index == len(expected_values, 'Too few values returned from range' 

Вот пример его использования:

generator_tester(
    Gen(...), 
    [1, 2, 3, 4, 5, 6, 7, 8, 9, 10] 
)