third-party/mruby/mrbgems/mruby-proc-ext/test/proc.rb

   1 ##
   2 # Proc(Ext) Test
   3
   4 assert('Proc#source_location') do
   5   loc = Proc.new {}.source_location
   6   next true if loc.nil?
   7   assert_equal loc[0][-7, 7], 'proc.rb'
   8   assert_equal loc[1], 5
   9 end
  10
  11 assert('Proc#inspect') do
  12   ins = Proc.new{}.inspect
  13   assert_kind_of String, ins
  14 end
  15
  16 assert('Proc#parameters') do
  17   parameters = Proc.new{|x,y=42,*other|}.parameters
  18   assert_equal [[:opt, :x], [:opt, :y], [:rest, :other]], parameters
  19 end
  20
  21 assert('Proc#lambda?') do
  22   assert_true lambda{}.lambda?
  23   assert_true !Proc.new{}.lambda?
  24 end
  25
  26 assert('Proc#===') do
  27   proc = Proc.new {|a| a * 2}
  28   assert_equal 20, (proc === 10)
  29 end
  30
  31 assert('Proc#yield') do
  32   proc = Proc.new {|a| a * 2}
  33   assert_equal 20, proc.yield(10)
  34 end
  35
  36 assert('Proc#curry') do
  37   b = proc {|x, y, z| (x||0) + (y||0) + (z||0) }
  38   assert_equal 6, b.curry[1][2][3]
  39   assert_equal 6, b.curry[1, 2][3, 4]
  40   assert_equal 6, b.curry(5)[1][2][3][4][5]
  41   assert_equal 6, b.curry(5)[1, 2][3, 4][5]
  42   assert_equal 1, b.curry(1)[1]
  43
  44   b = lambda {|x, y, z| (x||0) + (y||0) + (z||0) }
  45   assert_equal 6, b.curry[1][2][3]
  46   assert_raise(ArgumentError) { b.curry[1, 2][3, 4] }
  47   assert_raise(ArgumentError) { b.curry(5) }
  48   assert_raise(ArgumentError) { b.curry(1) }
  49
  50   assert_false(proc{}.curry.lambda?)
  51   assert_true(lambda{}.curry.lambda?)
  52 end
  53
  54 assert('Proc#parameters') do
  55   assert_equal([], Proc.new {}.parameters)
  56   assert_equal([], Proc.new {||}.parameters)
  57   assert_equal([[:opt, :a]], Proc.new {|a|}.parameters)
  58   assert_equal([[:req, :a]], lambda {|a|}.parameters)
  59   assert_equal([[:opt, :a]], lambda {|a=nil|}.parameters)
  60   assert_equal([[:req, :a]], ->(a){}.parameters)
  61   assert_equal([[:rest]], lambda { |*| }.parameters)
  62   assert_equal([[:rest, :a]], Proc.new {|*a|}.parameters)
  63   assert_equal([[:opt, :a], [:opt, :b], [:opt, :c], [:opt, :d], [:rest, :e], [:opt, :f], [:opt, :g], [:block, :h]], Proc.new {|a,b,c=:c,d=:d,*e,f,g,&h|}.parameters)
  64   assert_equal([[:req, :a], [:req, :b], [:opt, :c], [:opt, :d], [:rest, :e], [:req, :f], [:req, :g], [:block, :h]], lambda {|a,b,c=:c,d=:d,*e,f,g,&h|}.parameters)
  65 end
  66
  67 assert('Proc#to_proc') do
  68   proc = Proc.new {}
  69   assert_equal proc, proc.to_proc
  70 end
  71
  72 assert('Kernel#proc') do
  73   assert_true !proc{|a|}.lambda?
  74
  75   assert_raise LocalJumpError do
  76     proc{ break }.call
  77   end
  78 end
  79
  80 assert "Proc#<< and Proc#>>" do
  81   add3 = ->(n) { n + 3 }
  82   mul2 = ->(n) { n * 2 }
  83
  84   f1 = mul2 << add3
  85   assert_kind_of Proc, f1
  86   assert_equal 16, f1.call(5)
  87
  88   f2 = mul2 >> add3
  89   assert_kind_of Proc, f2
  90   assert_equal 13, f2.call(5)
  91 end
  92
  93 assert('mrb_proc_new_cfunc_with_env') do
  94   ProcExtTest.mrb_proc_new_cfunc_with_env(:test)
  95   ProcExtTest.mrb_proc_new_cfunc_with_env(:mruby)
  96
  97   t = ProcExtTest.new
  98
  99   assert_equal :test, t.test
 100   assert_equal :mruby, t.mruby
 101 end
 102
 103 assert('mrb_cfunc_env_get') do
 104   ProcExtTest.mrb_cfunc_env_get :get_int, [0, 1, 2]
 105
 106   t = ProcExtTest.new
 107
 108   assert_raise(TypeError) { t.cfunc_without_env }
 109
 110   assert_raise(IndexError) { t.get_int(-1) }
 111   assert_raise(IndexError) { t.get_int(3) }
 112
 113   assert_equal 1, t.get_int(1)
 114 end