[ruby/yarp] Fix rational parsing

https://github.com/ruby/yarp/commit/c8f31eb5b6

2 files changed, 16 insertions, 10 deletions

diff --git a/lib/yarp.rb b/lib/yarp.rb
index 5a176e9e62..f43aaa3d12 100644
--- a/lib/yarp.rb
+++ b/lib/yarp.rb
@@ -343,7 +343,7 @@ module YARP
 
   class RationalNode < Node
     def value
-      Rational(numeric.value)
+      Rational(slice.chomp("r"))
     end
   end
 
diff --git a/test/yarp/ruby_api_test.rb b/test/yarp/ruby_api_test.rb
index a26b971b82..1f66084ad2 100644
--- a/test/yarp/ruby_api_test.rb
+++ b/test/yarp/ruby_api_test.rb
@@ -8,7 +8,6 @@ class YARPRubyAPITest < Test::Unit::TestCase
     source = File.read(filepath, binmode: true, external_encoding: Encoding::UTF_8)
 
     assert_equal YARP.lex(source, filepath).value, YARP.lex_file(filepath).value
-
     assert_equal YARP.dump(source, filepath), YARP.dump_file(filepath)
 
     serialized = YARP.dump(source, filepath)
@@ -21,13 +20,20 @@ class YARPRubyAPITest < Test::Unit::TestCase
   end
 
   def test_literal_value_method
-    assert_equal 123, YARP.parse("123").value.statements.body.first.value
-    assert_equal 3.14, YARP.parse("3.14").value.statements.body.first.value
-    assert_equal 42i, YARP.parse("42i").value.statements.body.first.value
-    assert_equal 3.14i, YARP.parse("3.14i").value.statements.body.first.value
-    assert_equal 42r, YARP.parse("42r").value.statements.body.first.value
-    assert_equal 0.5r, YARP.parse("0.5r").value.statements.body.first.value
-    assert_equal 42ri, YARP.parse("42ri").value.statements.body.first.value
-    assert_equal 0.5ri, YARP.parse("0.5ri").value.statements.body.first.value
+    assert_equal 123, parse_expression("123").value
+    assert_equal 3.14, parse_expression("3.14").value
+    assert_equal 42i, parse_expression("42i").value
+    assert_equal 42.1ri, parse_expression("42.1ri").value
+    assert_equal 3.14i, parse_expression("3.14i").value
+    assert_equal 42r, parse_expression("42r").value
+    assert_equal 0.5r, parse_expression("0.5r").value
+    assert_equal 42ri, parse_expression("42ri").value
+    assert_equal 0.5ri, parse_expression("0.5ri").value
+  end
+
+  private
+
+  def parse_expression(source)
+    YARP.parse(source).value.statements.body.first
   end
 end