diff options
author | k0kubun <k0kubun@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2018-02-07 16:15:21 +0000 |
---|---|---|
committer | k0kubun <k0kubun@b2dd03c8-39d4-4d8f-98ff-823fe69b080e> | 2018-02-07 16:15:21 +0000 |
commit | 402e22832f0f863656c474c742439eeeff34e704 (patch) | |
tree | 47c03edc6734dd01494ff9fc27e84f435dd2b65b /test | |
parent | f5e1962400fff54295b9accecf1b92f1fd5f082b (diff) | |
download | ruby-402e22832f0f863656c474c742439eeeff34e704.tar.gz |
test_jit.rb: add initial test for JIT
git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@62291 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'test')
-rw-r--r-- | test/ruby/test_jit.rb | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/test/ruby/test_jit.rb b/test/ruby/test_jit.rb new file mode 100644 index 0000000000..85ce611067 --- /dev/null +++ b/test/ruby/test_jit.rb @@ -0,0 +1,58 @@ +# frozen_string_literal: true +require 'test/unit' + +# Test for --jit option +class TestJIT < Test::Unit::TestCase + JIT_TIMEOUT = 600 # 10min for each... + JIT_SUCCESS_PREFIX = 'JIT success \(\d+\.\dms\)' + + def test_jit + assert_eval_with_jit('print proc { 1 + 1 }.call', stdout: '2', success_count: 1) + end + + def test_jit_output + skip unless jit_available? + + out, err = eval_with_jit('5.times { puts "MJIT" }', verbose: 1, min_calls: 5) + assert_equal("MJIT\n" * 5, out) + assert_match(/^#{JIT_SUCCESS_PREFIX}: block in <main>@-e:1 -> .+_ruby_mjit_p\d+u\d+\.c$/, err) + assert_match(/^Successful MJIT finish$/, err) + end + + private + + # Shorthand for normal test cases + def assert_eval_with_jit(script, stdout: nil, success_count:) + out, err = eval_with_jit(script, verbose: 1, min_calls: 1) + actual = err.scan(/^#{JIT_SUCCESS_PREFIX}:/).size + assert_equal( + success_count, actual, + "Expected #{success_count} times of JIT success, but succeeded #{actual} times.\n\n"\ + "script:\n#{code_block(script)}\nstderr:\n#{code_block(err)}", + ) + if stdout + assert_match(stdout, out, "Expected stderr #{out.inspect} to match #{stdout.inspect} with script:\n#{code_block(script)}") + end + end + + # Run Ruby script with --jit-wait (Synchronous JIT compilation). + # Returns [stdout, stderr] + def eval_with_jit(script, verbose: 0, min_calls: 5) + stdout, stderr, status = EnvUtil.invoke_ruby( + ['--disable-gems', '--jit-wait', "--jit-verbose=#{verbose}", "--jit-min-calls=#{min_calls}", '-e', script], + '', true, true, timeout: JIT_TIMEOUT, + ) + assert_equal(true, status.success?, "Failed to run script with JIT:\n#{code_block(script)}") + [stdout, stderr] + end + + def code_block(code) + "```\n#{code}\n```\n\n" + end + + # If this is false, tests which require JIT should be skipped. + # When this is not checked, probably the test expects Ruby to behave in the same way even if JIT is not supported. + def jit_available? + @jit_available ||= (EnvUtil.invoke_ruby(['--jit', '-e', 'print RubyVM::MJIT.enabled?'], '', true, true).first == 'true') + end +end |