def test_step(self): def my_gen(): yield 1 yield 2 task = Task("t1", None) node = ExecNode(task, None) node.generator = my_gen() assert 1 == node.step() assert 2 == node.step() assert None == node.step()