[Truffle] Adding Array#select! to array.rb

jruby · Mar 23, 2015 · b696c67 · b696c67
1 parent 3ea7fcd
commit b696c67
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/spec/truffle/tags/core/array/select_tags.txt b/spec/truffle/tags/core/array/select_tags.txt
diff --git a/truffle/src/main/ruby/core/rubinius/common/array.rb b/truffle/src/main/ruby/core/rubinius/common/array.rb
@@ -926,6 +926,15 @@ def sample(count=undefined, options=undefined)
     return count == size ? result : result[0, count]
   end
 
+  def select!(&block)
+    return to_enum :select! unless block_given?
+
+    Rubinius.check_frozen
+
+    ary = select(&block)
+    replace ary unless size == ary.size
+  end
+
   def find_index(obj=undefined)
     super
   end