1berry/ruby
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[PRISM] Add raw option to assert_prism_eval

Browse Source
This commit is contained in:
Peter Zhu 2024-01-25 16:59:42 -05:00
parent 2034e6ad5a
commit 5bd6949151
1 changed files with 9 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
test/ruby/test_compile_prism.rb
View File

@ -59,10 +59,7 @@ module Prism
end end
def test_SourceLineNode def test_SourceLineNode
ruby_eval = RubyVM::InstructionSequence.compile("__LINE__").eval assert_prism_eval("__LINE__", raw: true)
prism_eval = RubyVM::InstructionSequence.compile_prism("__LINE__").eval
assert_equal ruby_eval, prism_eval
end end
def test_TrueNode def test_TrueNode
@ -776,10 +773,7 @@ module Prism
"hello".equal?("hello") "hello".equal?("hello")
RUBY RUBY
].each do |src| ].each do |src|
ruby_eval = RubyVM::InstructionSequence.compile(src).eval assert_prism_eval(src, raw: true)
prism_eval = RubyVM::InstructionSequence.compile_prism(src).eval
assert_equal ruby_eval, prism_eval, src
end end
end end
@ -1710,15 +1704,8 @@ end
end end
def test_PreExecutionNode def test_PreExecutionNode
# BEGIN {} must be defined at the top level, so we need to manually assert_prism_eval("BEGIN { a = 1 }; 2", raw: true)
# call the evals here instead of calling `assert_prism_eval` assert_prism_eval("b = 2; BEGIN { a = 1 }; a + b", raw: true)
ruby_eval = RubyVM::InstructionSequence.compile("BEGIN { a = 1 }; 2").eval
prism_eval = RubyVM::InstructionSequence.compile_prism("BEGIN { a = 1 }; 2").eval
assert_equal ruby_eval, prism_eval
ruby_eval = RubyVM::InstructionSequence.compile("b = 2; BEGIN { a = 1 }; a + b").eval
prism_eval = RubyVM::InstructionSequence.compile_prism("b = 2; BEGIN { a = 1 }; a + b").eval
assert_equal ruby_eval, prism_eval
end end
def test_PostExecutionNode def test_PostExecutionNode
@ -2498,8 +2485,8 @@ end
private private
def compare_eval(source) def compare_eval(source, raw:)
source = "class Prism::TestCompilePrism\n#{source}\nend" source = raw ? source : "class Prism::TestCompilePrism\n#{source}\nend"
ruby_eval = RubyVM::InstructionSequence.compile(source).eval ruby_eval = RubyVM::InstructionSequence.compile(source).eval
prism_eval = RubyVM::InstructionSequence.compile_prism(source).eval prism_eval = RubyVM::InstructionSequence.compile_prism(source).eval
@ -2511,14 +2498,14 @@ end
end end
end end
def assert_prism_eval(source) def assert_prism_eval(source, raw: false)
$VERBOSE, verbose_bak = nil, $VERBOSE $VERBOSE, verbose_bak = nil, $VERBOSE
begin begin
compare_eval(source) compare_eval(source, raw:)
# Test "popped" functionality # Test "popped" functionality
compare_eval("#{source}; 1") compare_eval("#{source}; 1", raw:)
ensure ensure
$VERBOSE = verbose_bak $VERBOSE = verbose_bak
end end