code.kerkeslager.com Git - fur/blob - util.py

   1 import functools
   2
   3 def force_generator(generator_function):
   4     @functools.wraps(generator_function)
   5     def forced_generator(*args, **kwargs):
   6         return list(generator_function(*args, **kwargs))
   7
   8     return forced_generator
   9
  10 if __name__ == '__main__':
  11     import unittest
  12
  13     class ForceGeneratorTests(unittest.TestCase):
  14         def test_forces_generator(self):
  15             forced_range = force_generator(range)
  16
  17             self.assertEqual(
  18                 forced_range(10),
  19                 [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
  20             )
  21
  22     unittest.main()