summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authormrkn <mrkn@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2016-05-17 13:58:46 +0000
committermrkn <mrkn@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>2016-05-17 13:58:46 +0000
commit41ef7ec381338a97d15a6b4b18acd8b426a9ce79 (patch)
tree7fca575e53a48e43b015df51e43e8fb8e26f6a62 /test
parent341b34d7339c267c51448d35d9e8f47413fbcc53 (diff)
Implement Enumerable#sum
* enum.c (enum_sum): Implement Enumerable#sum. * test/ruby/test_enum.rb (test_sum): Test sum for Enumerable. * test/ruby/test_hash.rb (test_sum): Test sum for Hash. * test/ruby/test_range.rb (test_sum): Test sum for Range. git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@55032 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
Diffstat (limited to 'test')
-rw-r--r--test/ruby/test_enum.rb90
-rw-r--r--test/ruby/test_hash.rb5
-rw-r--r--test/ruby/test_range.rb5
3 files changed, 100 insertions, 0 deletions
diff --git a/test/ruby/test_enum.rb b/test/ruby/test_enum.rb
index 4951cd1b2b..ed464a285e 100644
--- a/test/ruby/test_enum.rb
+++ b/test/ruby/test_enum.rb
@@ -815,4 +815,94 @@ class TestEnumerable < Test::Unit::TestCase
assert_equal([[1,2],0,[3,4],1],
@obj.each_with_index.flat_map(&lambda2))
end
+
+ def assert_typed_equal(e, v, cls, msg=nil)
+ assert_kind_of(cls, v, msg)
+ assert_equal(e, v, msg)
+ end
+
+ def assert_int_equal(e, v, msg=nil)
+ assert_typed_equal(e, v, Integer, msg)
+ end
+
+ def assert_rational_equal(e, v, msg=nil)
+ assert_typed_equal(e, v, Rational, msg)
+ end
+
+ def assert_float_equal(e, v, msg=nil)
+ assert_typed_equal(e, v, Float, msg)
+ end
+
+ def assert_complex_equal(e, v, msg=nil)
+ assert_typed_equal(e, v, Complex, msg)
+ end
+
+ def test_sum
+ class << (enum = Object.new)
+ include Enumerable
+ def each
+ yield 3
+ yield 5
+ yield 7
+ end
+ end
+ assert_int_equal(15, enum.sum)
+
+ assert_int_equal(0, [].each.sum)
+ assert_int_equal(3, [3].each.sum)
+ assert_int_equal(8, [3, 5].each.sum)
+ assert_int_equal(15, [3, 5, 7].each.sum)
+ assert_rational_equal(8r, [3, 5r].each.sum)
+ assert_float_equal(15.0, [3, 5, 7.0].each.sum)
+ assert_float_equal(15.0, [3, 5r, 7.0].each.sum)
+ assert_complex_equal(8r + 1i, [3, 5r, 1i].each.sum)
+ assert_complex_equal(15.0 + 1i, [3, 5r, 7.0, 1i].each.sum)
+
+ assert_int_equal(2*FIXNUM_MAX, Array.new(2, FIXNUM_MAX).each.sum)
+ assert_int_equal(2*(FIXNUM_MAX+1), Array.new(2, FIXNUM_MAX+1).each.sum)
+ assert_int_equal(10*FIXNUM_MAX, Array.new(10, FIXNUM_MAX).each.sum)
+ assert_int_equal(0, ([FIXNUM_MAX, 1, -FIXNUM_MAX, -1]*10).each.sum)
+ assert_int_equal(FIXNUM_MAX*10, ([FIXNUM_MAX+1, -1]*10).each.sum)
+ assert_int_equal(2*FIXNUM_MIN, Array.new(2, FIXNUM_MIN).each.sum)
+
+ assert_float_equal(0.0, [].each.sum(0.0))
+ assert_float_equal(3.0, [3].each.sum(0.0))
+ assert_float_equal(3.5, [3].each.sum(0.5))
+ assert_float_equal(8.5, [3.5, 5].each.sum)
+ assert_float_equal(10.5, [2, 8.5].each.sum)
+ assert_float_equal((FIXNUM_MAX+1).to_f, [FIXNUM_MAX, 1, 0.0].each.sum)
+ assert_float_equal((FIXNUM_MAX+1).to_f, [0.0, FIXNUM_MAX+1].each.sum)
+
+ assert_rational_equal(3/2r, [1/2r, 1].each.sum)
+ assert_rational_equal(5/6r, [1/2r, 1/3r].each.sum)
+
+ assert_equal(2.0+3.0i, [2.0, 3.0i].each.sum)
+
+ assert_int_equal(13, [1, 2].each.sum(10))
+ assert_int_equal(16, [1, 2].each.sum(10) {|v| v * 2 })
+
+ yielded = []
+ three = SimpleDelegator.new(3)
+ ary = [1, 2.0, three]
+ assert_float_equal(12.0, ary.each.sum {|x| yielded << x; x * 2 })
+ assert_equal(ary, yielded)
+
+ assert_raise(TypeError) { [Object.new].each.sum }
+
+ large_number = 100000000
+ small_number = 1e-9
+ until (large_number + small_number) == large_number
+ small_number /= 10
+ end
+ assert_float_equal(large_number+(small_number*10), [large_number, *[small_number]*10].each.sum)
+ assert_float_equal(large_number+(small_number*10), [large_number/1r, *[small_number]*10].each.sum)
+ assert_float_equal(large_number+(small_number*11), [small_number, large_number/1r, *[small_number]*10].each.sum)
+
+ assert_equal("abc", ["a", "b", "c"].each.sum(""))
+ assert_equal([1, [2], 3], [[1], [[2]], [3]].each.sum([]))
+
+ assert_separately(%w[-rmathn], <<-EOS, ignore_stderr: true)
+ assert_equal(6, [1r, 2, 3r].each.sum)
+ EOS
+ end
end
diff --git a/test/ruby/test_hash.rb b/test/ruby/test_hash.rb
index 6a8e7f0423..03a712d85b 100644
--- a/test/ruby/test_hash.rb
+++ b/test/ruby/test_hash.rb
@@ -1415,6 +1415,11 @@ class TestHash < Test::Unit::TestCase
assert_equal([10, 20, 30], [1, 2, 3].map(&h))
end
+ def test_sum
+ histogram = { 1 => 6, 2 => 4, 3 => 3, 4 => 7, 5 => 5, 6 => 4 }
+ assert_equal(100, histogram.sum {|v, n| v * n })
+ end
+
class TestSubHash < TestHash
class SubHash < Hash
def reject(*)
diff --git a/test/ruby/test_range.rb b/test/ruby/test_range.rb
index 2bff105936..431ed0dc30 100644
--- a/test/ruby/test_range.rb
+++ b/test/ruby/test_range.rb
@@ -631,4 +631,9 @@ class TestRange < Test::Unit::TestCase
end
(a.."c").each {|x, &b| assert_nil(b)}
end
+
+ def test_sum
+ assert_equal(55, (1..10).sum)
+ assert_equal(110, (1..10).sum {|v| v * 2 })
+ end
end