1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203
|
require 'test/unit'
class TestRubyOptimization < Test::Unit::TestCase
BIGNUM_POS_MIN_32 = 1073741824 # 2 ** 30
if BIGNUM_POS_MIN_32.kind_of?(Fixnum)
FIXNUM_MAX = 4611686018427387903 # 2 ** 62 - 1
else
FIXNUM_MAX = 1073741823 # 2 ** 30 - 1
end
BIGNUM_NEG_MAX_32 = -1073741825 # -2 ** 30 - 1
if BIGNUM_NEG_MAX_32.kind_of?(Fixnum)
FIXNUM_MIN = -4611686018427387904 # -2 ** 62
else
FIXNUM_MIN = -1073741824 # -2 ** 30
end
def redefine_method(klass, method)
(@redefine_method_seq ||= 0)
seq = (@redefine_method_seq += 1)
eval(<<-End, ::TOPLEVEL_BINDING)
class #{klass}
alias redefine_method_orig_#{seq} #{method}
undef #{method}
def #{method}(*args)
args[0]
end
end
End
begin
return yield
ensure
eval(<<-End, ::TOPLEVEL_BINDING)
class #{klass}
undef #{method}
alias #{method} redefine_method_orig_#{seq}
end
End
end
end
def test_fixnum_plus
a, b = 1, 2
assert_equal 3, a + b
assert_instance_of Fixnum, FIXNUM_MAX
assert_instance_of Bignum, FIXNUM_MAX + 1
assert_equal 21, 10 + 11
assert_equal 11, redefine_method('Fixnum', '+') { 10 + 11 }
assert_equal 21, 10 + 11
end
def test_fixnum_minus
assert_equal 5, 8 - 3
assert_instance_of Fixnum, FIXNUM_MIN
assert_instance_of Bignum, FIXNUM_MIN - 1
assert_equal 5, 8 - 3
assert_equal 3, redefine_method('Fixnum', '-') { 8 - 3 }
assert_equal 5, 8 - 3
end
def test_fixnum_mul
assert_equal 15, 3 * 5
end
def test_fixnum_div
assert_equal 3, 15 / 5
assert_equal 6.66, redefine_method('Float', '/') { 4.2 / 6.66 }, "bug 9238"
end
def test_fixnum_mod
assert_equal 1, 8 % 7
end
def test_float_plus
assert_equal 4.0, 2.0 + 2.0
assert_equal 2.0, redefine_method('Float', '+') { 2.0 + 2.0 }
assert_equal 4.0, 2.0 + 2.0
end
def test_string_length
assert_equal 6, "string".length
assert_nil redefine_method('String', 'length') { "string".length }
assert_equal 6, "string".length
end
def test_string_empty?
assert_equal true, "".empty?
assert_equal false, "string".empty?
assert_nil redefine_method('String', 'empty?') { "string".empty? }
assert_equal true, "".empty?
assert_equal false, "string".empty?
end
def test_string_plus
assert_equal "", "" + ""
assert_equal "x", "x" + ""
assert_equal "x", "" + "x"
assert_equal "ab", "a" + "b"
assert_equal 'b', redefine_method('String', '+') { "a" + "b" }
assert_equal "ab", "a" + "b"
end
def test_string_succ
assert_equal 'b', 'a'.succ
assert_equal 'B', 'A'.succ
end
def test_string_format
assert_equal '2', '%d' % 2
end
def test_array_plus
assert_equal [1,2], [1]+[2]
end
def test_array_minus
assert_equal [2], [1,2] - [1]
end
def test_array_length
assert_equal 0, [].length
assert_equal 3, [1,2,3].length
end
def test_array_empty?
assert_equal true, [].empty?
assert_equal false, [1,2,3].empty?
end
def test_hash_length
assert_equal 0, {}.length
assert_equal 1, {1=>1}.length
end
def test_hash_empty?
assert_equal true, {}.empty?
assert_equal false, {1=>1}.empty?
end
class MyObj
def ==(other)
true
end
end
def test_eq
assert_equal true, nil == nil
assert_equal true, 1 == 1
assert_equal true, 'string' == 'string'
assert_equal true, 1 == MyObj.new
assert_equal false, nil == MyObj.new
assert_equal true, MyObj.new == 1
assert_equal true, MyObj.new == nil
end
def test_tailcall
bug4082 = '[ruby-core:33289]'
option = {
tailcall_optimization: true,
trace_instruction: false,
}
iseq = RubyVM::InstructionSequence.new(<<-EOF, "Bug#4082", bug4082, nil, option).eval
class #{self.class}::Tailcall
def fact_helper(n, res)
if n == 1
res
else
fact_helper(n - 1, n * res)
end
end
def fact(n)
fact_helper(n, 1)
end
end
EOF
assert_equal(9131, Tailcall.new.fact(3000).to_s.size, bug4082)
end
def test_tailcall_with_block
bug6901 = '[ruby-dev:46065]'
option = {
tailcall_optimization: true,
trace_instruction: false,
}
iseq = RubyVM::InstructionSequence.new(<<-EOF, "Bug#6901", bug6901, nil, option).eval
def identity(val)
val
end
def delay
-> {
identity(yield)
}
end
EOF
assert_equal(123, delay { 123 }.call, bug6901)
end
end
|