def test_generator(self):
""" Test adding a generator """
context = Context()
self.assertEqual(0, len(context.get_generators()))
my_gen = count_gen()
context.add_generator('gen', my_gen)
self.assertEqual(1, len(context.get_generators()))
self.assertTrue('gen' in context.get_generators())
self.assertTrue(context.get_generator('gen') is not None)
def test_generator_bind(self):
""" Test generator setting to variables """
context = Context()
self.assertEqual(0, len(context.get_generators()))
my_gen = count_gen()
context.add_generator('gen', my_gen)
context.bind_generator_next('foo', 'gen')
self.assertEqual(1, context.mod_count)
self.assertEqual(1, context.get_value('foo'))
self.assertTrue(2, next(context.get_generator('gen')))
self.assertTrue(3, next(my_gen))