tests/run/generator_expressions.pyx

   1 # mode: run
   2 # tag: generators, lambda
   3
   4 def genexpr():
   5     """
   6     >>> genexpr()
   7     [0, 2, 4, 6, 8]
   8     """
   9     x = 'abc'
  10     result = list( x*2 for x in range(5) )
  11     assert x == 'abc' # don't leak
  12     return result
  13
  14 def genexpr_if():
  15     """
  16     >>> genexpr_if()
  17     [0, 4, 8]
  18     """
  19     x = 'abc'
  20     result = list( x*2 for x in range(5) if x % 2 == 0 )
  21     assert x == 'abc' # don't leak
  22     return result
  23
  24 def genexpr_with_lambda():
  25     """
  26     >>> genexpr_with_lambda()
  27     [0, 4, 8]
  28     """
  29     x = 'abc'
  30     result = list( x*2 for x in range(5) if (lambda x:x % 2)(x) == 0 )
  31     assert x == 'abc' # don't leak
  32     return result
  33
  34 def genexpr_of_lambdas(int N):
  35     """
  36     >>> [ (f(), g()) for f,g in genexpr_of_lambdas(5) ]
  37     [(0, 0), (1, 2), (2, 4), (3, 6), (4, 8)]
  38     """
  39     return ( ((lambda : x), (lambda : x*2)) for x in range(N) )