[GR-19220] Enumerable:multiple yielded values; pass and untag (#2145)

PullRequest: truffleruby/2146

Author: eregon

spec/tags/core/enumerable/count_tags.txt

@@ -114,7 +114,10 @@ def count(item = undefined)
     seq = 0
     if !Primitive.undefined?(item)
       warn 'given block not used', uplevel: 1 if block_given?
-      each { |o| seq += 1 if item == o }
+      each do
+        o = Primitive.single_block_arg
+        seq += 1 if item == o
+      end
     elsif block_given?
       each { |o| seq += 1 if yield(o) }
     else
@@ -973,7 +976,8 @@ def uniq(&block)
     result = []
     if block_given?
       h = {}
-      each do |e|
+      each do
+        e = Primitive.single_block_arg
         v = yield(e)
         unless h.key?(v)
           h[v] = true
@@ -982,7 +986,8 @@ def uniq(&block)
       end
     else
       h = {}
-      each do |e|
+      each do
+        e = Primitive.single_block_arg
         unless h.key?(e)
           h[e] = true
           result << e