Merge pull request #568 from kostya/partition

array#partition
topazproject · Mar 31, 2013 · 720a707d0617793b2520bbfe765d62ad80dc081a · 720a707
2 parents 9167847 + 1d11e9a
commit 720a707d0617793b2520bbfe765d62ad80dc081a
Unified Split

Showing with 9 additions and 6 deletions.

+9 −0 lib-topaz/enumerable.rb

+0 −4 spec/tags/core/array/partition_tags.txt

+0 −2 spec/tags/core/enumerable/partition_tags.txt
diff --git a/lib-topaz/enumerable.rb b/lib-topaz/enumerable.rb
@@ -176,6 +176,15 @@ def max
     end
   end
 
+  def partition(&block)
+    return self.enum_for(:partition) unless block
+    a, b = [], []
+    self.each do |e|
+      block.call(e) ? a.push(e) : b.push(e)
+    end
+    [a, b]
+  end
+
   def count(*args, &block)
     c = 0
     if args.empty?

diff --git a/spec/tags/core/array/partition_tags.txt b/spec/tags/core/array/partition_tags.txt
diff --git a/spec/tags/core/enumerable/partition_tags.txt b/spec/tags/core/enumerable/partition_tags.txt
@@ -1,3 +1 @@
-fails:Enumerable#partition returns two arrays, the first containing elements for which the block is true, the second containing the rest
-fails:Enumerable#partition returns an Enumerator if called without a block
 fails:Enumerable#partition gathers whole arrays as elements when each yields multiple