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, context.get_generator('gen').next()) self.assertTrue(3, my_gen.next())