Merge pull request #473 from OpakAlex/fix-integer-parsing-for-yaml

fix parsing integer values with '_' at the end

1 files changed, 13 insertions, 2 deletions

diff --git a/test/psych/test_scalar_scanner.rb b/test/psych/test_scalar_scanner.rb
index 1bd6488..2cd129e 100644
--- a/test/psych/test_scalar_scanner.rb
+++ b/test/psych/test_scalar_scanner.rb
@@ -111,7 +111,15 @@ module Psych
     end
 
     def test_scan_strings_starting_with_underscores
-      assert_equal "_100", ss.tokenize('_100')
+      assert_equal '_100', ss.tokenize('_100')
+    end
+
+    def test_scan_strings_starting_with_number
+      assert_equal '450D', ss.tokenize('450D')
+    end
+
+    def test_scan_strings_ending_with_underscores
+      assert_equal '100_', ss.tokenize('100_')
     end
 
     def test_scan_int_commas_and_underscores
@@ -120,7 +128,10 @@ module Psych
       assert_equal 123_456_789, ss.tokenize('123_456_789')
       assert_equal 123_456_789, ss.tokenize('123,456,789')
       assert_equal 123_456_789, ss.tokenize('1_2,3,4_5,6_789')
-      assert_equal 123_456_789, ss.tokenize('1_2,3,4_5,6_789_')
+
+      assert_equal 1, ss.tokenize('1')
+      assert_equal 1, ss.tokenize('+1')
+      assert_equal -1, ss.tokenize('-1')
 
       assert_equal 0b010101010, ss.tokenize('0b010101010')
       assert_equal 0b010101010, ss.tokenize('0b0,1_0,1_,0,1_01,0')