Syntax-highlight yield in IRB

lib/irb/color.rb

@@ -25,7 +25,7 @@ module IRB # :nodoc:
       on_ident:           [[BLUE, BOLD],            [Ripper::EXPR_ENDFN]],
       on_int:             [[BLUE, BOLD],            [Ripper::EXPR_END]],
       on_float:           [[MAGENTA, BOLD],         [Ripper::EXPR_END]],
-      on_kw:              [[GREEN],                 [Ripper::EXPR_CLASS, Ripper::EXPR_BEG, Ripper::EXPR_END, Ripper::EXPR_FNAME]],
+      on_kw:              [[GREEN],                 [Ripper::EXPR_ARG, Ripper::EXPR_CLASS, Ripper::EXPR_BEG, Ripper::EXPR_END, Ripper::EXPR_FNAME]],
       on_label:           [[MAGENTA],               [Ripper::EXPR_LABELED]],
       on_qwords_beg:      [[RED],                   [Ripper::EXPR_BEG]],
       on_regexp_beg:      [[RED, BOLD],             [Ripper::EXPR_BEG]],

test/irb/test_color.rb

@@ -24,6 +24,7 @@ module TestIRB
         'ERB.new("a#{nil}b", trim_mode: "-")' => "#{BLUE}#{BOLD}#{UNDERLINE}ERB#{CLEAR}.new(#{RED}\"#{CLEAR}#{RED}a#{CLEAR}#{RED}\#{#{CLEAR}#{CYAN}#{BOLD}nil#{CLEAR}#{RED}}#{CLEAR}#{RED}b#{CLEAR}#{RED}\"#{CLEAR}, #{MAGENTA}trim_mode:#{CLEAR} #{RED}\"#{CLEAR}#{RED}-#{CLEAR}#{RED}\"#{CLEAR})",
         "# comment" => "#{BLUE}#{BOLD}# comment#{CLEAR}",
         '/r#{e}g/' => "#{RED}#{BOLD}/#{CLEAR}#{RED}r#{CLEAR}#{RED}\#{#{CLEAR}e}#{RED}g#{CLEAR}#{RED}#{BOLD}/#{CLEAR}",
+        "yield(hello)" => "#{GREEN}yield#{CLEAR}(hello)",
       }.each do |code, result|
         assert_equal(result, with_term { IRB::Color.colorize_code(code) })
       end