require 'test/unit' require 'continuation' class TestHash < Test::Unit::TestCase def test_hash x = {1=>2, 2=>4, 3=>6} y = {1=>2, 2=>4, 3=>6} # y = {1, 2, 2, 4, 3, 6} # 1.9 doesn't support assert_equal(2, x[1]) assert(begin for k,v in y raise if k*2 != v end true rescue false end) assert_equal(3, x.length) assert(x.has_key?(1)) assert(x.has_value?(4)) assert_equal([4,6], x.values_at(2,3)) assert_equal({1=>2, 2=>4, 3=>6}, x) z = y.keys.join(":") assert_equal("1:2:3", z) z = y.values.join(":") assert_equal("2:4:6", z) assert_equal(x, y) y.shift assert_equal(2, y.length) z = [1,2] y[z] = 256 assert_equal(256, y[z]) x = Hash.new(0) x[1] = 1 assert_equal(1, x[1]) assert_equal(0, x[2]) x = Hash.new([]) assert_equal([], x[22]) assert_same(x[22], x[22]) x = Hash.new{[]} assert_equal([], x[22]) assert_not_same(x[22], x[22]) x = Hash.new{|h,kk| z = kk; h[kk] = kk*2} z = 0 assert_equal(44, x[22]) assert_equal(22, z) z = 0 assert_equal(44, x[22]) assert_equal(0, z) x.default = 5 assert_equal(5, x[23]) x = Hash.new def x.default(k) $z = k self[k] = k*2 end $z = 0 assert_equal(44, x[22]) assert_equal(22, $z) $z = 0 assert_equal(44, x[22]) assert_equal(0, $z) end # From rubicon def setup @cls = Hash @h = @cls[ 1 => 'one', 2 => 'two', 3 => 'three', self => 'self', true => 'true', nil => 'nil', 'nil' => nil ] @verbose = $VERBOSE $VERBOSE = nil end def teardown $VERBOSE = @verbose end def test_s_AREF h = @cls["a" => 100, "b" => 200] assert_equal(100, h['a']) assert_equal(200, h['b']) assert_nil(h['c']) h = @cls.[]("a" => 100, "b" => 200) assert_equal(100, h['a']) assert_equal(200, h['b']) assert_nil(h['c']) end def test_s_new h = @cls.new assert_instance_of(@cls, h) assert_nil(h.default) assert_nil(h['spurious']) h = @cls.new('default') assert_instance_of(@cls, h) assert_equal('default', h.default) assert_equal('default', h['spurious']) end def test_AREF # '[]' t = Time.now h = @cls[ 1 => 'one', 2 => 'two', 3 => 'three', self => 'self', t => 'time', nil => 'nil', 'nil' => nil ] assert_equal('one', h[1]) assert_equal('two', h[2]) assert_equal('three', h[3]) assert_equal('self', h[self]) assert_equal('time', h[t]) assert_equal('nil', h[nil]) assert_equal(nil, h['nil']) assert_equal(nil, h['koala']) h1 = h.dup h1.default = :default assert_equal('one', h1[1]) assert_equal('two', h1[2]) assert_equal('three', h1[3]) assert_equal('self', h1[self]) assert_equal('time', h1[t]) assert_equal('nil', h1[nil]) assert_equal(nil, h1['nil']) assert_equal(:default, h1['koala']) end def test_ASET # '[]=' t = Time.now h = @cls.new h[1] = 'one' h[2] = 'two' h[3] = 'three' h[self] = 'self' h[t] = 'time' h[nil] = 'nil' h['nil'] = nil assert_equal('one', h[1]) assert_equal('two', h[2]) assert_equal('three', h[3]) assert_equal('self', h[self]) assert_equal('time', h[t]) assert_equal('nil', h[nil]) assert_equal(nil, h['nil']) assert_equal(nil, h['koala']) h[1] = 1 h[nil] = 99 h['nil'] = nil z = [1,2] h[z] = 256 assert_equal(1, h[1]) assert_equal('two', h[2]) assert_equal('three', h[3]) assert_equal('self', h[self]) assert_equal('time', h[t]) assert_equal(99, h[nil]) assert_equal(nil, h['nil']) assert_equal(nil, h['koala']) assert_equal(256, h[z]) end def test_EQUAL # '==' h1 = @cls[ "a" => 1, "c" => 2 ] h2 = @cls[ "a" => 1, "c" => 2, 7 => 35 ] h3 = @cls[ "a" => 1, "c" => 2, 7 => 35 ] h4 = @cls[ ] assert(h1 == h1) assert(h2 == h2) assert(h3 == h3) assert(h4 == h4) assert(!(h1 == h2)) assert(h2 == h3) assert(!(h3 == h4)) end def test_clear assert(@h.size > 0) @h.clear assert_equal(0, @h.size) assert_nil(@h[1]) end def test_clone for taint in [ false, true ] for untrust in [ false, true ] for frozen in [ false, true ] a = @h.clone a.taint if taint a.untrust if untrust a.freeze if frozen b = a.clone assert_equal(a, b) assert(a.__id__ != b.__id__) assert_equal(a.frozen?, b.frozen?) assert_equal(a.untrusted?, b.untrusted?) assert_equal(a.tainted?, b.tainted?) end end end end def test_default assert_nil(@h.default) h = @cls.new(:xyzzy) assert_equal(:xyzzy, h.default) end def test_default= assert_nil(@h.default) @h.default = :xyzzy assert_equal(:xyzzy, @h.default) end def test_delete h1 = @cls[ 1 => 'one', 2 => 'two', true => 'true' ] h2 = @cls[ 1 => 'one', 2 => 'two' ] h3 = @cls[ 2 => 'two' ] assert_equal('true', h1.delete(true)) assert_equal(h2, h1) assert_equal('one', h1.delete(1)) assert_equal(h3, h1) assert_equal('two', h1.delete(2)) assert_equal(@cls[], h1) assert_nil(h1.delete(99)) assert_equal(@cls[], h1) assert_equal('default 99', h1.delete(99) {|i| "default #{i}" }) end def test_delete_if base = @cls[ 1 => 'one', 2 => false, true => 'true', 'cat' => 99 ] h1 = @cls[ 1 => 'one', 2 => false, true => 'true' ] h2 = @cls[ 2 => false, 'cat' => 99 ] h3 = @cls[ 2 => false ] h = base.dup assert_equal(h, h.delete_if { false }) assert_equal(@cls[], h.delete_if { true }) h = base.dup assert_equal(h1, h.delete_if {|k,v| k.instance_of?(String) }) assert_equal(h1, h) h = base.dup assert_equal(h2, h.delete_if {|k,v| v.instance_of?(String) }) assert_equal(h2, h) h = base.dup assert_equal(h3, h.delete_if {|k,v| v }) assert_equal(h3, h) h = base.dup n = 0 h.delete_if {|*a| n += 1 assert_equal(2, a.size) assert_equal(base[a[0]], a[1]) h.shift true } assert_equal(base.size, n) end def test_keep_if h = {1=>2,3=>4,5=>6} assert_equal({3=>4,5=>6}, h.keep_if {|k, v| k + v >= 7 }) h = {1=>2,3=>4,5=>6} assert_equal({1=>2,3=>4,5=>6}, h.keep_if{true}) end def test_dup for taint in [ false, true ] for untrust in [ false, true ] for frozen in [ false, true ] a = @h.dup a.taint if taint a.freeze if frozen b = a.dup assert_equal(a, b) assert(a.__id__ != b.__id__) assert_equal(false, b.frozen?) assert_equal(a.tainted?, b.tainted?) assert_equal(a.untrusted?, b.untrusted?) end end end end def test_each count = 0 @cls[].each { |k, v| count + 1 } assert_equal(0, count) h = @h h.each do |k, v| assert_equal(v, h.delete(k)) end assert_equal(@cls[], h) end def test_each_key count = 0 @cls[].each_key { |k| count + 1 } assert_equal(0, count) h = @h h.each_key do |k| h.delete(k) end assert_equal(@cls[], h) end def test_each_pair count = 0 @cls[].each_pair { |k, v| count + 1 } assert_equal(0, count) h = @h h.each_pair do |k, v| assert_equal(v, h.delete(k)) end assert_equal(@cls[], h) end def test_each_value res = [] @cls[].each_value { |v| res << v } assert_equal(0, [].length) @h.each_value { |v| res << v } assert_equal(0, [].length) expected = [] @h.each { |k, v| expected << v } assert_equal([], expected - res) assert_equal([], res - expected) end def test_empty? assert(@cls[].empty?) assert(!@h.empty?) end def test_fetch assert_raise(KeyError) { @cls[].fetch(1) } assert_raise(KeyError) { @h.fetch('gumby') } assert_equal('gumbygumby', @h.fetch('gumby') {|k| k * 2 }) assert_equal('pokey', @h.fetch('gumby', 'pokey')) assert_equal('one', @h.fetch(1)) assert_equal(nil, @h.fetch('nil')) assert_equal('nil', @h.fetch(nil)) end def test_key2? assert(!@cls[].key?(1)) assert(!@cls[].key?(nil)) assert(@h.key?(nil)) assert(@h.key?(1)) assert(!@h.key?('gumby')) end def test_value? assert(!@cls[].value?(1)) assert(!@cls[].value?(nil)) assert(@h.value?('one')) assert(@h.value?(nil)) assert(!@h.value?('gumby')) end def test_include? assert(!@cls[].include?(1)) assert(!@cls[].include?(nil)) assert(@h.include?(nil)) assert(@h.include?(1)) assert(!@h.include?('gumby')) end def test_key assert_equal(1, @h.key('one')) assert_equal(nil, @h.key('nil')) assert_equal('nil', @h.key(nil)) assert_equal(nil, @h.key('gumby')) assert_equal(nil, @cls[].key('gumby')) end def test_values_at res = @h.values_at('dog', 'cat', 'horse') assert(res.length == 3) assert_equal([nil, nil, nil], res) res = @h.values_at assert(res.length == 0) res = @h.values_at(3, 2, 1, nil) assert_equal 4, res.length assert_equal %w( three two one nil ), res res = @h.values_at(3, 99, 1, nil) assert_equal 4, res.length assert_equal ['three', nil, 'one', 'nil'], res end def test_invert h = @h.invert assert_equal(1, h['one']) assert_equal(true, h['true']) assert_equal(nil, h['nil']) h.each do |k, v| assert(@h.key?(v)) # not true in general, but works here end h = @cls[ 'a' => 1, 'b' => 2, 'c' => 1].invert assert_equal(2, h.length) assert(h[1] == 'a' || h[1] == 'c') assert_equal('b', h[2]) end def test_key? assert(!@cls[].key?(1)) assert(!@cls[].key?(nil)) assert(@h.key?(nil)) assert(@h.key?(1)) assert(!@h.key?('gumby')) end def test_keys assert_equal([], @cls[].keys) keys = @h.keys expected = [] @h.each { |k, v| expected << k } assert_equal([], keys - expected) assert_equal([], expected - keys) end def test_length assert_equal(0, @cls[].length) assert_equal(7, @h.length) end def test_member? assert(!@cls[].member?(1)) assert(!@cls[].member?(nil)) assert(@h.member?(nil)) assert(@h.member?(1)) assert(!@h.member?('gumby')) end def test_rehash a = [ "a", "b" ] c = [ "c", "d" ] h = @cls[ a => 100, c => 300 ] assert_equal(100, h[a]) a[0] = "z" assert_nil(h[a]) h.rehash assert_equal(100, h[a]) end def test_reject base = @cls[ 1 => 'one', 2 => false, true => 'true', 'cat' => 99 ] h1 = @cls[ 1 => 'one', 2 => false, true => 'true' ] h2 = @cls[ 2 => false, 'cat' => 99 ] h3 = @cls[ 2 => false ] h = base.dup assert_equal(h, h.reject { false }) assert_equal(@cls[], h.reject { true }) h = base.dup assert_equal(h1, h.reject {|k,v| k.instance_of?(String) }) assert_equal(h2, h.reject {|k,v| v.instance_of?(String) }) assert_equal(h3, h.reject {|k,v| v }) assert_equal(base, h) end def test_reject! base = @cls[ 1 => 'one', 2 => false, true => 'true', 'cat' => 99 ] h1 = @cls[ 1 => 'one', 2 => false, true => 'true' ] h2 = @cls[ 2 => false, 'cat' => 99 ] h3 = @cls[ 2 => false ] h = base.dup assert_equal(nil, h.reject! { false }) assert_equal(@cls[], h.reject! { true }) h = base.dup assert_equal(h1, h.reject! {|k,v| k.instance_of?(String) }) assert_equal(h1, h) h = base.dup assert_equal(h2, h.reject! {|k,v| v.instance_of?(String) }) assert_equal(h2, h) h = base.dup assert_equal(h3, h.reject! {|k,v| v }) assert_equal(h3, h) end def test_replace h = @cls[ 1 => 2, 3 => 4 ] h1 = h.replace(@cls[ 9 => 8, 7 => 6 ]) assert_equal(h, h1) assert_equal(8, h[9]) assert_equal(6, h[7]) assert_nil(h[1]) assert_nil(h[2]) end def test_shift h = @h.dup @h.length.times { k, v = h.shift assert(@h.key?(k)) assert_equal(@h[k], v) } assert_equal(0, h.length) end def test_size assert_equal(0, @cls[].length) assert_equal(7, @h.length) end def test_sort h = @cls[].sort assert_equal([], h) h = @cls[ 1 => 1, 2 => 1 ].sort assert_equal([[1,1], [2,1]], h) h = @cls[ 'cat' => 'feline', 'ass' => 'asinine', 'bee' => 'beeline' ] h1 = h.sort assert_equal([ %w(ass asinine), %w(bee beeline), %w(cat feline)], h1) end def test_store t = Time.now h = @cls.new h.store(1, 'one') h.store(2, 'two') h.store(3, 'three') h.store(self, 'self') h.store(t, 'time') h.store(nil, 'nil') h.store('nil', nil) assert_equal('one', h[1]) assert_equal('two', h[2]) assert_equal('three', h[3]) assert_equal('self', h[self]) assert_equal('time', h[t]) assert_equal('nil', h[nil]) assert_equal(nil, h['nil']) assert_equal(nil, h['koala']) h.store(1, 1) h.store(nil, 99) h.store('nil', nil) assert_equal(1, h[1]) assert_equal('two', h[2]) assert_equal('three', h[3]) assert_equal('self', h[self]) assert_equal('time', h[t]) assert_equal(99, h[nil]) assert_equal(nil, h['nil']) assert_equal(nil, h['koala']) end def test_to_a assert_equal([], @cls[].to_a) assert_equal([[1,2]], @cls[ 1=>2 ].to_a) a = @cls[ 1=>2, 3=>4, 5=>6 ].to_a assert_equal([1,2], a.delete([1,2])) assert_equal([3,4], a.delete([3,4])) assert_equal([5,6], a.delete([5,6])) assert_equal(0, a.length) h = @cls[ 1=>2, 3=>4, 5=>6 ] h.taint h.untrust a = h.to_a assert_equal(true, a.tainted?) assert_equal(true, a.untrusted?) end def test_to_hash h = @h.to_hash assert_equal(@h, h) end def test_to_s h = @cls[ 1 => 2, "cat" => "dog", 1.5 => :fred ] assert_equal(h.inspect, h.to_s) $, = ":" assert_equal(h.inspect, h.to_s) h = @cls[] assert_equal(h.inspect, h.to_s) ensure $, = nil end def test_update h1 = @cls[ 1 => 2, 2 => 3, 3 => 4 ] h2 = @cls[ 2 => 'two', 4 => 'four' ] ha = @cls[ 1 => 2, 2 => 'two', 3 => 4, 4 => 'four' ] hb = @cls[ 1 => 2, 2 => 3, 3 => 4, 4 => 'four' ] assert_equal(ha, h1.update(h2)) assert_equal(ha, h1) h1 = @cls[ 1 => 2, 2 => 3, 3 => 4 ] h2 = @cls[ 2 => 'two', 4 => 'four' ] assert_equal(hb, h2.update(h1)) assert_equal(hb, h2) end def test_value2? assert(!@cls[].value?(1)) assert(!@cls[].value?(nil)) assert(@h.value?(nil)) assert(@h.value?('one')) assert(!@h.value?('gumby')) end def test_values assert_equal([], @cls[].values) vals = @h.values expected = [] @h.each { |k, v| expected << v } assert_equal([], vals - expected) assert_equal([], expected - vals) end def test_security_check h = {} assert_raise(SecurityError) do Thread.new do $SAFE = 4 h[1] = 1 end.join end end def test_intialize_wrong_arguments assert_raise(ArgumentError) do Hash.new(0) { } end end def test_create assert_equal({1=>2, 3=>4}, Hash[[[1,2],[3,4]]]) assert_raise(ArgumentError) { Hash[0, 1, 2] } assert_equal({1=>2, 3=>4}, Hash[1,2,3,4]) o = Object.new def o.to_hash() {1=>2} end assert_equal({1=>2}, Hash[o], "[ruby-dev:34555]") end def test_rehash2 h = {1 => 2, 3 => 4} assert_equal(h.dup, h.rehash) assert_raise(RuntimeError) { h.each { h.rehash } } assert_equal({}, {}.rehash) end def test_fetch2 assert_equal(:bar, @h.fetch(0, :foo) { :bar }) end def test_default_proc h = Hash.new {|hh, k| hh + k + "baz" } assert_equal("foobarbaz", h.default_proc.call("foo", "bar")) h = {} assert_nil(h.default_proc) end def test_shift2 h = Hash.new {|hh, k| :foo } h[1] = 2 assert_equal([1, 2], h.shift) assert_equal(:foo, h.shift) assert_equal(:foo, h.shift) h = Hash.new(:foo) h[1] = 2 assert_equal([1, 2], h.shift) assert_equal(:foo, h.shift) assert_equal(:foo, h.shift) h = {1=>2} h.each { assert_equal([1, 2], h.shift) } end def test_reject_bang2 assert_equal({1=>2}, {1=>2,3=>4}.reject! {|k, v| k + v == 7 }) assert_nil({1=>2,3=>4}.reject! {|k, v| k == 5 }) assert_nil({}.reject! { }) end def test_select assert_equal({3=>4,5=>6}, {1=>2,3=>4,5=>6}.select {|k, v| k + v >= 7 }) end def test_select! h = {1=>2,3=>4,5=>6} assert_equal(h, h.select! {|k, v| k + v >= 7 }) assert_equal({3=>4,5=>6}, h) h = {1=>2,3=>4,5=>6} assert_equal(nil, h.select!{true}) end def test_clear2 assert_equal({}, {1=>2,3=>4,5=>6}.clear) h = {1=>2,3=>4,5=>6} h.each { h.clear } assert_equal({}, h) end def test_replace2 h1 = Hash.new { :foo } h2 = {} h2.replace h1 assert_equal(:foo, h2[0]) assert_raise(ArgumentError) { h2.replace() } assert_raise(TypeError) { h2.replace(1) } h2.freeze assert_raise(ArgumentError) { h2.replace() } assert_raise(RuntimeError) { h2.replace(h1) } assert_raise(RuntimeError) { h2.replace(42) } end def test_size2 assert_equal(0, {}.size) end def test_equal2 assert({} != 0) o = Object.new def o.to_hash; {}; end def o.==(x); true; end assert({} == o) def o.==(x); false; end assert({} != o) h1 = {1=>2}; h2 = {3=>4} assert(h1 != h2) h1 = {1=>2}; h2 = {1=>4} assert(h1 != h2) end def test_eql assert(!({}.eql?(0))) o = Object.new def o.to_hash; {}; end def o.eql?(x); true; end assert({}.eql?(o)) def o.eql?(x); false; end assert(!({}.eql?(o))) end def test_hash2 assert_kind_of(Integer, {}.hash) end def test_update2 h1 = {1=>2, 3=>4} h2 = {1=>3, 5=>7} h1.update(h2) {|k, v1, v2| k + v1 + v2 } assert_equal({1=>6, 3=>4, 5=>7}, h1) end def test_merge h1 = {1=>2, 3=>4} h2 = {1=>3, 5=>7} assert_equal({1=>3, 3=>4, 5=>7}, h1.merge(h2)) assert_equal({1=>6, 3=>4, 5=>7}, h1.merge(h2) {|k, v1, v2| k + v1 + v2 }) end def test_assoc assert_equal([3,4], {1=>2, 3=>4, 5=>6}.assoc(3)) assert_nil({1=>2, 3=>4, 5=>6}.assoc(4)) end def test_rassoc assert_equal([3,4], {1=>2, 3=>4, 5=>6}.rassoc(4)) assert_nil({1=>2, 3=>4, 5=>6}.rassoc(3)) end def test_flatten assert_equal([[1], [2]], {[1] => [2]}.flatten) end def test_callcc h = {1=>2} c = nil f = false h.each { callcc {|c2| c = c2 } } unless f f = true c.call end assert_raise(RuntimeError) { h.each { h.rehash } } h = {1=>2} c = nil assert_raise(RuntimeError) do h.each { callcc {|c2| c = c2 } } h.clear c.call end end def test_compare_by_identity a = "foo" assert(!{}.compare_by_identity?) h = { a => "bar" } assert(!h.compare_by_identity?) h.compare_by_identity assert(h.compare_by_identity?) #assert_equal("bar", h[a]) assert_nil(h["foo"]) end class ObjWithHash def initialize(value, hash) @value = value @hash = hash end attr_reader :value, :hash def eql?(other) @value == other.value end end def test_hash_hash assert_equal({0=>2,11=>1}.hash, {11=>1,0=>2}.hash) o1 = ObjWithHash.new(0,1) o2 = ObjWithHash.new(11,1) assert_equal({o1=>1,o2=>2}.hash, {o2=>2,o1=>1}.hash) end def test_hash_bignum_hash x = 2<<(32-3)-1 assert_equal({x=>1}.hash, {x=>1}.hash) x = 2<<(64-3)-1 assert_equal({x=>1}.hash, {x=>1}.hash) o = Object.new def o.hash; 2<<100; end assert_equal({x=>1}.hash, {x=>1}.hash) end def test_hash_poped assert_nothing_raised { eval("a = 1; {a => a}; a") } end def test_recursive_key h = {} assert_nothing_raised { h[h] = :foo } h.rehash assert_equal(:foo, h[h]) end end