Add a convenience value method for numeric literals

Author: eregon

lib/yarp.rb

@@ -313,6 +313,30 @@ def pretty_print(q)
     end
   end
 
+  class FloatNode < Node
+    def value
+      Float(slice)
+    end
+  end
+
+  class ImaginaryNode < Node
+    def value
+      Complex(0, numeric.value)
+    end
+  end
+
+  class IntegerNode < Node
+    def value
+      Integer(slice)
+    end
+  end
+
+  class RationalNode < Node
+    def value
+      Rational(numeric.value)
+    end
+  end
+
   # Load the serialized AST using the source as a reference into a tree.
   def self.load(source, serialized)
     Serialize.load(source, serialized)

test/ruby_api_test.rb

@@ -19,4 +19,15 @@ def test_ruby_api
     assert_equal_nodes ast1, ast2
     assert_equal_nodes ast2, ast3
   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
+  end
 end