enumerable_mapper 0.5.0 → 0.5.5

data/lib/enumerable_mapper/version.rb

@@ -2,7 +2,7 @@ module EnumerableMapper #:nodoc:
 	module VERSION #:nodoc:
 		MAJOR = 0
 		MINOR = 5
-		TINY  = 0
+		TINY  = 5
 
 		STRING = [MAJOR, MINOR, TINY].join('.')
 	end

data/lib/enumerable_mapper.rb

@@ -23,4 +23,16 @@ module Enumerable
 		Mapper.new( self )
 	end
 	alias_method :collected, :mapped
+
+	# Group collection according to the method name or block given.
+	#
+	# Examples:
+	#   Person.find( :all ).grouped_by( :company )
+	#   Person.find( :all ).grouped_by{ |p| p.born_on.year }
+	def grouped_by( method_id=nil )
+		raise "No method or block given" unless method_id or block_given?
+		groupings = self.map{ |i| ( block_given? ) ? yield( i ) : i.send( method_id ) }.uniq
+		groupings.map{ |g| [ g, self.select{ |i| ( ( block_given? ) ? yield( i ) : i.send( method_id ) ) == g } ] }
+	end
+
 end

data/test/test_grouped_by.rb

@@ -0,0 +1,42 @@
+require File.dirname(__FILE__) + '/test_helper.rb'
+
+class Foo
+	attr_accessor :name
+	attr_accessor :type
+	def initialize( name, type )
+		@name, @type = name, type
+	end
+end
+
+
+class TestGroupedBy < Test::Unit::TestCase
+
+	def setup
+	end
+
+	def create_array
+		[ Foo.new( 'inge', 'person' ), Foo.new( 'charlie', 'person' ), Foo.new( 'charlie', 'dog' ), Foo.new( 'oscar', 'dog' ) ]
+	end
+
+	def test_by_method_id
+		test_array = create_array
+		assert_equal 2,        test_array.grouped_by( :type ).length
+		assert_equal 3,        test_array.grouped_by( :name ).length
+		assert_equal 'person', test_array.grouped_by( :type )[0].first
+		assert_equal 'dog',    test_array.grouped_by( :type )[1].first
+		assert_equal 2,        test_array.grouped_by( :type )[0].last.length
+	end
+	
+	def test_by_block
+		test_array = create_array
+		assert_equal 2,        test_array.grouped_by{ |i| i.type }.length
+		assert_equal 3,        test_array.grouped_by{ |i| i.name }.length
+		assert_equal 'PERSON', test_array.grouped_by{ |i| i.type.upcase }[0].first
+	end
+	
+	def test_error
+		test_array = create_array
+		assert_raises( RuntimeError ) { test_array.grouped_by }
+	end
+	
+end

metadata

@@ -3,8 +3,8 @@ rubygems_version: 0.9.4
 specification_version: 1
 name: enumerable_mapper
 version: !ruby/object:Gem::Version 
-  version: 0.5.0
-date: 2007-08-29 00:00:00 +02:00
+  version: 0.5.5
+date: 2007-12-05 00:00:00 +01:00
 summary: This gem provides a shorthand for simple .map/.collect calls on collections.
 require_paths: 
 - lib
@@ -49,6 +49,7 @@ files:
 - test/test_helper.rb
 test_files: 
 - test/test_enumerable_mapper.rb
+- test/test_grouped_by.rb
 - test/test_helper.rb
 rdoc_options: 
 - --main