1## 2# Array(Ext) Test 3 4def assert_permutation_combination(exp, receiver, meth, *args) 5 act = [] 6 ret = receiver.__send__(meth, *args) { |v| act << v } 7 assert "assert_#{meth}" do 8 assert_equal(exp, act.sort) 9 assert_same(receiver, ret) 10 end 11end 12 13def assert_permutation(exp, receiver, *args) 14 assert_permutation_combination(exp, receiver, :permutation, *args) 15end 16 17def assert_combination(exp, receiver, *args) 18 assert_permutation_combination(exp, receiver, :combination, *args) 19end 20 21assert("Array#assoc") do 22 s1 = [ "colors", "red", "blue", "green" ] 23 s2 = [ "letters", "a", "b", "c" ] 24 s3 = "foo" 25 a = [ s1, s2, s3 ] 26 27 assert_equal [ "letters", "a", "b", "c" ], a.assoc("letters") 28 assert_nil a.assoc("foo") 29end 30 31assert("Array#at") do 32 a = [ "a", "b", "c", "d", "e" ] 33 assert_equal "a", a.at(0) 34 assert_equal "e", a.at(-1) 35end 36 37assert("Array#rassoc") do 38 a = [ [ 1, "one"], [2, "two"], [3, "three"], ["ii", "two"] ] 39 40 assert_equal [2, "two"], a.rassoc("two") 41 assert_nil a.rassoc("four") 42end 43 44assert("Array#uniq!") do 45 a = [1, 2, 3, 1] 46 a.uniq! 47 assert_equal [1, 2, 3], a 48 49 b = [ "a", "b", "c" ] 50 assert_nil b.uniq! 51 52 c = [["student","sam"], ["student","george"], ["teacher","matz"]] 53 assert_equal [["student", "sam"], ["teacher", "matz"]], c.uniq! { |s| s.first } 54 55 d = [["student","sam"], ["teacher","matz"]] 56 assert_nil d.uniq! { |s| s.first } 57end 58 59assert("Array#uniq") do 60 a = [1, 2, 3, 1] 61 assert_equal [1, 2, 3], a.uniq 62 assert_equal [1, 2, 3, 1], a 63 64 b = [["student","sam"], ["student","george"], ["teacher","matz"]] 65 assert_equal [["student", "sam"], ["teacher", "matz"]], b.uniq { |s| s.first } 66end 67 68assert("Array#-") do 69 a = [1, 2, 3, 1] 70 b = [1] 71 c = 1 72 73 assert_raise(TypeError) { a - c } 74 assert_equal [2, 3], (a - b) 75 assert_equal [1, 2, 3, 1], a 76end 77 78assert("Array#|") do 79 a = [1, 2, 3, 1] 80 b = [1, 4] 81 c = 1 82 83 assert_raise(TypeError) { a | c } 84 assert_equal [1, 2, 3, 4], (a | b) 85 assert_equal [1, 2, 3, 1], a 86end 87 88assert("Array#union") do 89 a = [1, 2, 3, 1] 90 b = [1, 4] 91 c = [1, 5] 92 93 assert_equal [1, 2, 3, 4, 5], a.union(b,c) 94end 95 96assert("Array#difference") do 97 a = [1, 2, 3, 1, 6, 7] 98 b = [1, 4, 6] 99 c = [1, 5, 7] 100 101 assert_equal [2, 3], a.difference(b,c) 102end 103 104assert("Array#&") do 105 a = [1, 2, 3, 1] 106 b = [1, 4] 107 c = 1 108 109 assert_raise(TypeError) { a & c } 110 assert_equal [1], (a & b) 111 assert_equal [1, 2, 3, 1], a 112end 113 114assert("Array#intersection") do 115 a = [1, 2, 3, 1, 8, 6, 7, 8] 116 b = [1, 4, 6, 8] 117 c = [1, 5, 7, 8] 118 119 assert_equal [1, 8], a.intersection(b,c) 120end 121 122assert("Array#flatten") do 123 assert_equal [1, 2, "3", {4=>5}, :'6'], [1, 2, "3", {4=>5}, :'6'].flatten 124 assert_equal [1, 2, 3, 4, 5, 6], [1, 2, [3, 4, 5], 6].flatten 125 assert_equal [1, 2, 3, 4, 5, 6], [1, 2, [3, [4, 5], 6]].flatten 126 assert_equal [1, [2, [3, [4, [5, [6]]]]]], [1, [2, [3, [4, [5, [6]]]]]].flatten(0) 127 assert_equal [1, 2, [3, [4, [5, [6]]]]], [1, [2, [3, [4, [5, [6]]]]]].flatten(1) 128 assert_equal [1, 2, 3, [4, [5, [6]]]], [1, [2, [3, [4, [5, [6]]]]]].flatten(2) 129 assert_equal [1, 2, 3, 4, [5, [6]]], [1, [2, [3, [4, [5, [6]]]]]].flatten(3) 130 assert_equal [1, 2, 3, 4, 5, [6]], [1, [2, [3, [4, [5, [6]]]]]].flatten(4) 131 assert_equal [1, 2, 3, 4, 5, 6], [1, [2, [3, [4, [5, [6]]]]]].flatten(5) 132end 133 134assert("Array#flatten!") do 135 assert_equal [1, 2, 3, 4, 5, 6], [1, 2, [3, [4, 5], 6]].flatten! 136end 137 138assert("Array#compact") do 139 a = [1, nil, "2", nil, :t, false, nil] 140 assert_equal [1, "2", :t, false], a.compact 141 assert_equal [1, nil, "2", nil, :t, false, nil], a 142end 143 144assert("Array#compact!") do 145 a = [1, nil, "2", nil, :t, false, nil] 146 a.compact! 147 assert_equal [1, "2", :t, false], a 148end 149 150assert("Array#fetch") do 151 a = [ 11, 22, 33, 44 ] 152 assert_equal 22, a.fetch(1) 153 assert_equal 44, a.fetch(-1) 154 assert_equal 'cat', a.fetch(4, 'cat') 155 ret = 0 156 a.fetch(100) { |i| ret = i } 157 assert_equal 100, ret 158 assert_raise(IndexError) { a.fetch(100) } 159end 160 161assert("Array#fill") do 162 a = [ "a", "b", "c", "d" ] 163 assert_equal ["x", "x", "x", "x"], a.fill("x") 164 assert_equal ["x", "x", "x", "w"], a.fill("w", -1) 165 assert_equal ["x", "x", "z", "z"], a.fill("z", 2, 2) 166 assert_equal ["y", "y", "z", "z"], a.fill("y", 0..1) 167 assert_equal [0, 1, 4, 9], a.fill { |i| i*i } 168 assert_equal [0, 1, 8, 27], a.fill(-2) { |i| i*i*i } 169 assert_equal [0, 2, 3, 27], a.fill(1, 2) { |i| i+1 } 170 assert_equal [1, 2, 3, 27], a.fill(0..1) { |i| i+1 } 171 assert_raise(ArgumentError) { a.fill } 172 173 assert_equal([0, 1, 2, 3, -1, 5], [0, 1, 2, 3, 4, 5].fill(-1, -2, 1)) 174 assert_equal([0, 1, 2, 3, -1, -1, -1], [0, 1, 2, 3, 4, 5].fill(-1, -2, 3)) 175 assert_equal([0, 1, 2, -1, -1, 5], [0, 1, 2, 3, 4, 5].fill(-1, 3..4)) 176 assert_equal([0, 1, 2, -1, 4, 5], [0, 1, 2, 3, 4, 5].fill(-1, 3...4)) 177 assert_equal([0, 1, -1, -1, -1, 5], [0, 1, 2, 3, 4, 5].fill(-1, 2..-2)) 178 assert_equal([0, 1, -1, -1, 4, 5], [0, 1, 2, 3, 4, 5].fill(-1, 2...-2)) 179 assert_equal([0, 1, 2, 13, 14, 5], [0, 1, 2, 3, 4, 5].fill(3..4){|i| i+10}) 180 assert_equal([0, 1, 2, 13, 4, 5], [0, 1, 2, 3, 4, 5].fill(3...4){|i| i+10}) 181 assert_equal([0, 1, 12, 13, 14, 5], [0, 1, 2, 3, 4, 5].fill(2..-2){|i| i+10}) 182 assert_equal([0, 1, 12, 13, 4, 5], [0, 1, 2, 3, 4, 5].fill(2...-2){|i| i+10}) 183 184 assert_equal [1, 2, 3, 4, 'x', 'x'], [1, 2, 3, 4, 5, 6].fill('x', -2..-1) 185 assert_equal [1, 2, 3, 4, 'x', 6], [1, 2, 3, 4, 5, 6].fill('x', -2...-1) 186 assert_equal [1, 2, 3, 4, 5, 6], [1, 2, 3, 4, 5, 6].fill('x', -2...-2) 187 assert_equal [1, 2, 3, 4, 'x', 6], [1, 2, 3, 4, 5, 6].fill('x', -2..-2) 188 assert_equal [1, 2, 3, 4, 5, 6], [1, 2, 3, 4, 5, 6].fill('x', -2..0) 189end 190 191assert("Array#reverse_each") do 192 a = [ "a", "b", "c", "d" ] 193 b = [] 194 a.reverse_each do |i| 195 b << i 196 end 197 assert_equal [ "d", "c", "b", "a" ], b 198end 199 200assert("Array#rotate") do 201 a = ["a", "b", "c", "d"] 202 assert_equal ["b", "c", "d", "a"], a.rotate 203 assert_equal ["a", "b", "c", "d"], a 204 assert_equal ["c", "d", "a", "b"], a.rotate(2) 205 assert_equal ["b", "c", "d", "a"], a.rotate(-3) 206 assert_equal ["c", "d", "a", "b"], a.rotate(10) 207 assert_equal [], [].rotate 208end 209 210assert("Array#rotate!") do 211 a = ["a", "b", "c", "d"] 212 assert_equal ["b", "c", "d", "a"], a.rotate! 213 assert_equal ["b", "c", "d", "a"], a 214 assert_equal ["d", "a", "b", "c"], a.rotate!(2) 215 assert_equal ["a", "b", "c", "d"], a.rotate!(-3) 216 assert_equal ["c", "d", "a", "b"], a.rotate(10) 217 assert_equal [], [].rotate! 218end 219 220assert("Array#delete_if") do 221 a = [1, 2, 3, 4, 5] 222 assert_equal [1, 2, 3, 4, 5], a.delete_if { false } 223 assert_equal [1, 2, 3, 4, 5], a 224 225 a = [1, 2, 3, 4, 5] 226 assert_equal [], a.delete_if { true } 227 assert_equal [], a 228 229 a = [1, 2, 3, 4, 5] 230 assert_equal [1, 2, 3], a.delete_if { |i| i > 3 } 231 assert_equal [1, 2, 3], a 232end 233 234assert("Array#reject!") do 235 a = [1, 2, 3, 4, 5] 236 assert_nil a.reject! { false } 237 assert_equal [1, 2, 3, 4, 5], a 238 239 a = [1, 2, 3, 4, 5] 240 assert_equal [], a.reject! { true } 241 assert_equal [], a 242 243 a = [1, 2, 3, 4, 5] 244 assert_equal [1, 2, 3], a.reject! { |val| val > 3 } 245 assert_equal [1, 2, 3], a 246end 247 248assert("Array#insert") do 249 a = ["a", "b", "c", "d"] 250 assert_equal ["a", "b", 99, "c", "d"], a.insert(2, 99) 251 assert_equal ["a", "b", 99, "c", 1, 2, 3, "d"], a.insert(-2, 1, 2, 3) 252 253 b = ["a", "b", "c", "d"] 254 assert_equal ["a", "b", "c", "d", nil, nil, 99], b.insert(6, 99) 255end 256 257assert("Array#bsearch") do 258 # Find minimum mode 259 a = [0, 2, 4] 260 assert_equal 0, a.bsearch{ |x| x >= -1 } 261 assert_equal 0, a.bsearch{ |x| x >= 0 } 262 assert_equal 2, a.bsearch{ |x| x >= 1 } 263 assert_equal 2, a.bsearch{ |x| x >= 2 } 264 assert_equal 4, a.bsearch{ |x| x >= 3 } 265 assert_equal 4, a.bsearch{ |x| x >= 4 } 266 assert_nil a.bsearch{ |x| x >= 5 } 267 268 # Find any mode 269 a = [0, 4, 8] 270 def between(lo, x, hi) 271 if x < lo 272 1 273 elsif x > hi 274 -1 275 else 276 0 277 end 278 end 279 assert_nil a.bsearch{ |x| between(-3, x, -1) } 280 assert_equal 0, a.bsearch{ |x| between(-1, x, 1) } 281 assert_nil a.bsearch{ |x| between( 1, x, 3) } 282 assert_equal 4, a.bsearch{ |x| between( 3, x, 5) } 283 assert_nil a.bsearch{ |x| between( 5, x, 7) } 284 assert_equal 8, a.bsearch{ |x| between( 7, x, 9) } 285 assert_nil a.bsearch{ |x| between( 9, x, 11) } 286 287 assert_equal 0, a.bsearch{ |x| between( 0, x, 3) } 288 assert_equal 4, a.bsearch{ |x| between( 0, x, 4) } 289 assert_equal 4, a.bsearch{ |x| between( 4, x, 8) } 290 assert_equal 8, a.bsearch{ |x| between( 5, x, 8) } 291 292 # Invalid block result 293 assert_raise TypeError, 'invalid block result (must be numeric, true, false or nil)' do 294 a.bsearch{ 'I like to watch the world burn' } 295 end 296end 297 298# tested through Array#bsearch 299#assert("Array#bsearch_index") do 300#end 301 302assert("Array#keep_if") do 303 a = [1, 2, 3, 4, 5] 304 assert_equal [1, 2, 3, 4, 5], a.keep_if { true } 305 assert_equal [1, 2, 3, 4, 5], a 306 307 a = [1, 2, 3, 4, 5] 308 assert_equal [], a.keep_if { false } 309 assert_equal [], a 310 311 a = [1, 2, 3, 4, 5] 312 assert_equal [4, 5], a.keep_if { |val| val > 3 } 313 assert_equal [4, 5], a 314end 315 316assert("Array#select!") do 317 a = [1, 2, 3, 4, 5] 318 assert_nil a.select! { true } 319 assert_equal [1, 2, 3, 4, 5], a 320 321 a = [1, 2, 3, 4, 5] 322 assert_equal [], a.select! { false } 323 assert_equal [], a 324 325 a = [1, 2, 3, 4, 5] 326 assert_equal [4, 5], a.select! { |val| val > 3 } 327 assert_equal [4, 5], a 328end 329 330assert('Array#values_at') do 331 a = %w{red green purple white none} 332 333 assert_equal %w{red purple none}, a.values_at(0, 2, 4) 334 assert_equal ['green', 'white', nil, nil], a.values_at(1, 3, 5, 7) 335 assert_equal ['none', 'white', 'white', nil], a.values_at(-1, -2, -2, -7) 336 assert_equal ['none', nil, nil, 'red', 'green', 'purple'], a.values_at(4..6, 0...3) 337 assert_raise(TypeError) { a.values_at 'tt' } 338end 339 340assert('Array#to_h') do 341 assert_equal({}, [].to_h) 342 assert_equal({a: 1, b:2}, [[:a, 1], [:b, 2]].to_h) 343 344 assert_raise(TypeError) { [1].to_h } 345 assert_raise(ArgumentError) { [[1]].to_h } 346end 347 348assert("Array#index (block)") do 349 assert_nil (1..10).to_a.index { |i| i % 5 == 0 and i % 7 == 0 } 350 assert_equal 34, (1..100).to_a.index { |i| i % 5 == 0 and i % 7 == 0 } 351end 352 353assert("Array#dig") do 354 h = [[[1]], 0] 355 assert_equal(1, h.dig(0, 0, 0)) 356 assert_nil(h.dig(2, 0)) 357 assert_raise(TypeError) {h.dig(:a)} 358end 359 360assert("Array#slice!") do 361 a = [1, 2, 3] 362 b = a.slice!(0) 363 c = [1, 2, 3, 4, 5] 364 d = c.slice!(0, 2) 365 e = [1, 2, 3, 4, 5] 366 f = e.slice!(1..3) 367 g = [1, 2, 3] 368 h = g.slice!(-1) 369 i = [1, 2, 3] 370 j = i.slice!(0, -1) 371 372 assert_equal(a, [2, 3]) 373 assert_equal(b, 1) 374 assert_equal(c, [3, 4, 5]) 375 assert_equal(d, [1, 2]) 376 assert_equal(e, [1, 5]) 377 assert_equal(f, [2, 3, 4]) 378 assert_equal(g, [1, 2]) 379 assert_equal(h, 3) 380 assert_equal(i, [1, 2, 3]) 381 assert_equal(j, nil) 382end 383 384assert("Array#permutation") do 385 a = [1, 2, 3] 386 assert_permutation([[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]], a) 387 assert_permutation([[1],[2],[3]], a, 1) 388 assert_permutation([[1,2],[1,3],[2,1],[2,3],[3,1],[3,2]], a, 2) 389 assert_permutation([[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]], a, 3) 390 assert_permutation([[]], a, 0) 391 assert_permutation([], a, 4) 392 assert_permutation([], a, -1) 393end 394 395assert("Array#combination") do 396 a = [1, 2, 3, 4] 397 assert_combination([[1],[2],[3],[4]], a, 1) 398 assert_combination([[1,2],[1,3],[1,4],[2,3],[2,4],[3,4]], a, 2) 399 assert_combination([[1,2,3],[1,2,4],[1,3,4],[2,3,4]], a, 3) 400 assert_combination([[1,2,3,4]], a, 4) 401 assert_combination([[]], a, 0) 402 assert_combination([], a, 5) 403 assert_combination([], a, -1) 404end 405 406assert('Array#transpose') do 407 assert_equal([].transpose, []) 408 assert_equal([[]].transpose, []) 409 assert_equal([[1]].transpose, [[1]]) 410 assert_equal([[1,2,3]].transpose, [[1], [2], [3]]) 411 assert_equal([[1], [2], [3]].transpose, [[1,2,3]]) 412 assert_equal([[1,2], [3,4], [5,6]].transpose, [[1,3,5], [2,4,6]]) 413 assert_raise(TypeError) { [1].transpose } 414 assert_raise(IndexError) { [[1], [2,3,4]].transpose } 415end 416