예제 #1
0
 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()
예제 #2
0
 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()