mug 0.2.5 → 0.2.6

data/lib/mug.rb

@@ -3,11 +3,13 @@ require_relative 'mug/and-or'
 require_relative 'mug/apply'
 require_relative 'mug/array/extend'
 require_relative 'mug/bool'
+require_relative 'mug/counts'
 require_relative 'mug/fragile-method-chain'
 require_relative 'mug/hash/map'
 require_relative 'mug/hash/operations'
 require_relative 'mug/iterator/for'
 require_relative 'mug/iterator/method'
+require_relative 'mug/loop-with'
 require_relative 'mug/maybe'
 require_relative 'mug/self'
 require_relative 'mug/tau'

data/lib/mug/counts.rb

@@ -0,0 +1,34 @@
+
+module Enumerable
+
+	#
+	# Returns a hash of item=>count showing how many
+	# of each +item+ are in this Enumerable.
+	#
+	def counts &block
+		return counts_by(&block) if block_given?
+		hsh = Hash.new{|h,k| h[k] = 0 }
+		each do |k|
+			hsh[k] += 1
+		end
+		hsh
+	end
+
+	#
+	# Passes each element in turn to the block, and returns a
+	# hash of result=>count.
+	#
+	# If no block is given, an enumerator is returned.
+	#
+	def counts_by &block
+		return enum_for(:counts_by) unless block_given?
+		hsh = Hash.new{|h,k| h[k] = 0 }
+		each do |j|
+			k = yield j
+			hsh[k] += 1
+		end
+		hsh
+	end
+
+end
+

data/lib/mug/loop-with.rb

@@ -0,0 +1,34 @@
+
+module Kernel
+	#
+	# Repeatedly executes the block, yielding the current iteration
+	# count, which starts from +offset+. If no block is given, returns
+	# a new Enumerator that includes the iteration count, starting
+	# from +offset+
+	#
+	def loop_with_index(offset=0)
+		return c=enum_for(:loop_with_index, offset) unless block_given?
+		c = 0 + offset
+		while true
+			yield c
+			c += 1
+		end
+	rescue StopIteration
+	ensure
+		return c
+	end
+
+	#
+	# Repeatedly executes the block, yielding an arbitrary object, +obj+.
+	#
+	def loop_with_object(obj)
+		return obj=enum_for(:loop_with_object, obj) unless block_given?
+		while true
+			yield obj
+		end
+	rescue StopIteration
+	ensure
+		return obj
+	end
+end
+

data/test/test-counts.rb

@@ -0,0 +1,28 @@
+require 'test/unit'
+$VERBOSE = true
+
+require_relative '../lib/mug/counts'
+class Test_hashmap < Test::Unit::TestCase
+	def test_counts
+		a = %w(a b b c c c)
+		c = {'a'=>1, 'b'=>2, 'c'=>3}
+		assert_equal( c, a.counts )
+	end
+	def test_counts__block
+		a = %w(a b b c c c)
+		c = {'A'=>1, 'B'=>2, 'C'=>3}
+		assert_equal( c, a.counts{|i|i.upcase} )
+	end
+	def test_counts_by
+		a = %w(a b b c c c)
+		c = {'A'=>1, 'B'=>2, 'C'=>3}
+		assert_equal( c, a.counts_by{|i|i.upcase} )
+	end
+	def test_counts_by__noblock
+		a = %w(a b b c c c)
+		c = {'A'=>1, 'B'=>2, 'C'=>3}
+		e = a.counts_by
+		assert_equal( c, e.each{|i|i.upcase} )
+	end
+end
+

data/test/test-loop-with.rb

@@ -0,0 +1,75 @@
+require 'test/unit'
+$VERBOSE = true
+
+require_relative '../lib/mug/loop-with'
+class Test_loop_with < Test::Unit::TestCase
+
+	def test_loop_with_index__block
+		a = []
+		b = []
+		x = loop_with_index do |i|
+			a << i
+			break if a.length >= 3
+		end
+		y = loop_with_index do |i|
+			b << i
+			throw StopIteration if b.length >= 3
+		end
+		assert_equal( [0,1,2], a )
+		assert_equal( [0,1,2], b )
+		assert_equal( 2, x )
+		assert_equal( 2, y )
+	end
+	def test_loop_with_index__block_offset
+		a = []
+		b = []
+		x = loop_with_index(10) do |i|
+			a << i
+			break if a.length >= 3
+		end
+		y = loop_with_index(10) do |i|
+			b << i
+			throw StopIteration if b.length >= 3
+		end
+		assert_equal( [10,11,12], a )
+		assert_equal( [10,11,12], b )
+		assert_equal( 12, x )
+		assert_equal( 12, y )
+	end
+	def test_loop_with_index__enum
+		a = []
+		b = []
+		enum = loop_with_index
+		x = enum.each do |i|
+			a << i
+			break if a.length >= 3
+		end
+		y = enum.each do |i|
+			b << i
+			throw StopIteration if b.length >= 3
+		end
+		assert_equal( [0,1,2], a )
+		assert_equal( [0,1,2], b )
+		assert_equal( 2, x )
+		assert_equal( 2, y )
+	end
+	def test_loop_with_index__enum_offset
+		a = []
+		b = []
+		enum = loop_with_index(10)
+		x = enum.each do |i|
+			a << i
+			break if a.length >= 3
+		end
+		y = enum.each do |i|
+			b << i
+			throw StopIteration if b.length >= 3
+		end
+		assert_equal( [10,11,12], a )
+		assert_equal( [10,11,12], b )
+		assert_equal( 12, x )
+		assert_equal( 12, y )
+	end
+
+end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mug
 version: !ruby/object:Gem::Version
-  version: 0.2.5
+  version: 0.2.6
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-10-08 00:00:00.000000000 Z
+date: 2014-12-01 00:00:00.000000000 Z
 dependencies: []
 description: ! '== MUG: Matty''s Ultimate Gem
 
@@ -41,8 +41,10 @@ files:
 - lib/mug/apply.rb
 - lib/mug/to_h.rb
 - lib/mug/array/extend.rb
+- lib/mug/loop-with.rb
 - lib/mug/fragile-method-chain.rb
 - lib/mug/hashmap.rb
+- lib/mug/counts.rb
 - lib/mug/iterator_c.rb
 - lib/mug/hashop.rb
 - lib/mug/iterator.rb
@@ -55,11 +57,13 @@ files:
 - test/test_apply.rb
 - test/test-bool.rb
 - test/test-top.rb
+- test/test-loop-with.rb
 - test/test-to_h.rb
 - test/test-array-extend.rb
 - test/test-self.rb
 - test/test-iterator-method.rb
 - test/test-maybe.rb
+- test/test-counts.rb
 - test/test-fragile-method-chain.rb
 homepage: http://phluid61.github.com/mug
 licenses:
@@ -95,10 +99,12 @@ test_files:
 - test/test_apply.rb
 - test/test-bool.rb
 - test/test-top.rb
+- test/test-loop-with.rb
 - test/test-to_h.rb
 - test/test-array-extend.rb
 - test/test-self.rb
 - test/test-iterator-method.rb
 - test/test-maybe.rb
+- test/test-counts.rb
 - test/test-fragile-method-chain.rb
 has_rdoc: true