PageRenderTime 69ms CodeModel.GetById 10ms RepoModel.GetById 0ms app.codeStats 1ms

/test/ruby/test_array.rb

http://github.com/ruby/ruby
Ruby | 3231 lines | 2712 code | 477 blank | 42 comment | 49 complexity | 915f1b674d3b594dfee8a6c4cd5ee3c5 MD5 | raw file
Possible License(s): GPL-2.0, BSD-3-Clause, AGPL-3.0
  1. # coding: US-ASCII
  2. # frozen_string_literal: false
  3. require 'test/unit'
  4. require "delegate"
  5. require "rbconfig/sizeof"
  6. class TestArray < Test::Unit::TestCase
  7. def setup
  8. @verbose = $VERBOSE
  9. $VERBOSE = nil
  10. @cls = Array
  11. end
  12. def teardown
  13. $VERBOSE = @verbose
  14. end
  15. def test_percent_i
  16. assert_equal([:foo, :bar], %i[foo bar])
  17. assert_equal([:"\"foo"], %i["foo])
  18. end
  19. def test_percent_I
  20. x = 10
  21. assert_equal([:foo, :b10], %I[foo b#{x}])
  22. assert_equal([:"\"foo10"], %I["foo#{x}])
  23. end
  24. def test_0_literal
  25. assert_equal([1, 2, 3, 4], [1, 2] + [3, 4])
  26. assert_equal([1, 2, 1, 2], [1, 2] * 2)
  27. assert_equal("1:2", [1, 2] * ":")
  28. assert_equal([1, 2].hash, [1, 2].hash)
  29. assert_equal([2,3], [1,2,3] & [2,3,4])
  30. assert_equal([1,2,3,4], [1,2,3] | [2,3,4])
  31. assert_equal([1,2,3] - [2,3], [1])
  32. x = [0, 1, 2, 3, 4, 5]
  33. assert_equal(2, x[2])
  34. assert_equal([1, 2, 3], x[1..3])
  35. assert_equal([1, 2, 3], x[1,3])
  36. assert_equal([3, 4, 5], x[3..])
  37. assert_equal([0, 1, 2], x[..2])
  38. assert_equal([0, 1], x[...2])
  39. x[0, 2] = 10
  40. assert_equal([10, 2, 3, 4, 5], x)
  41. x[0, 0] = -1
  42. assert_equal([-1, 10, 2, 3, 4, 5], x)
  43. x[-1, 1] = 20
  44. assert_equal(20, x[-1])
  45. assert_equal(20, x.pop)
  46. end
  47. def test_array_andor_0
  48. assert_equal([2], ([1,2,3]&[2,4,6]))
  49. assert_equal([1,2,3,4,6], ([1,2,3]|[2,4,6]))
  50. end
  51. def test_compact_0
  52. a = [nil, 1, nil, nil, 5, nil, nil]
  53. assert_equal [1, 5], a.compact
  54. assert_equal [nil, 1, nil, nil, 5, nil, nil], a
  55. a.compact!
  56. assert_equal [1, 5], a
  57. end
  58. def test_uniq_0
  59. x = [1, 1, 4, 2, 5, 4, 5, 1, 2]
  60. x.uniq!
  61. assert_equal([1, 4, 2, 5], x)
  62. end
  63. def test_empty_0
  64. assert_equal true, [].empty?
  65. assert_equal false, [1].empty?
  66. assert_equal false, [1, 1, 4, 2, 5, 4, 5, 1, 2].empty?
  67. end
  68. def test_sort_0
  69. x = ["it", "came", "to", "pass", "that", "..."]
  70. x = x.sort.join(" ")
  71. assert_equal("... came it pass that to", x)
  72. x = [2,5,3,1,7]
  73. x.sort!{|a,b| a<=>b} # sort with condition
  74. assert_equal([1,2,3,5,7], x)
  75. x.sort!{|a,b| b-a} # reverse sort
  76. assert_equal([7,5,3,2,1], x)
  77. end
  78. def test_split_0
  79. x = "The Book of Mormon"
  80. assert_equal(x.reverse, x.split(//).reverse!.join)
  81. assert_equal(x.reverse, x.reverse!)
  82. assert_equal("g:n:i:r:t:s: :e:t:y:b: :1", "1 byte string".split(//).reverse.join(":"))
  83. x = "a b c d"
  84. assert_equal(['a', 'b', 'c', 'd'], x.split)
  85. assert_equal(['a', 'b', 'c', 'd'], x.split(' '))
  86. end
  87. def test_misc_0
  88. assert(defined? "a".chomp, '"a".chomp is not defined')
  89. assert_equal(["a", "b", "c"], "abc".scan(/./))
  90. assert_equal([["1a"], ["2b"], ["3c"]], "1a2b3c".scan(/(\d.)/))
  91. # non-greedy match
  92. assert_equal([["a", "12"], ["b", "22"]], "a=12;b=22".scan(/(.*?)=(\d*);?/))
  93. x = [1]
  94. assert_equal('1:1:1:1:1', (x * 5).join(":"))
  95. assert_equal('1', (x * 1).join(":"))
  96. assert_equal('', (x * 0).join(":"))
  97. *x = *(1..7).to_a
  98. assert_equal(7, x.size)
  99. assert_equal([1, 2, 3, 4, 5, 6, 7], x)
  100. x = [1,2,3]
  101. x[1,0] = x
  102. assert_equal([1,1,2,3,2,3], x)
  103. x = [1,2,3]
  104. x[-1,0] = x
  105. assert_equal([1,2,1,2,3,3], x)
  106. x = [1,2,3]
  107. x.concat(x)
  108. assert_equal([1,2,3,1,2,3], x)
  109. x = [1,2,3]
  110. x.clear
  111. assert_equal([], x)
  112. x = [1,2,3]
  113. y = x.dup
  114. x << 4
  115. y << 5
  116. assert_equal([1,2,3,4], x)
  117. assert_equal([1,2,3,5], y)
  118. end
  119. def test_beg_end_0
  120. x = [1, 2, 3, 4, 5]
  121. assert_equal(1, x.first)
  122. assert_equal([1], x.first(1))
  123. assert_equal([1, 2, 3], x.first(3))
  124. assert_raise_with_message(ArgumentError, /0\.\.1/) {x.first(1, 2)}
  125. assert_equal(5, x.last)
  126. assert_equal([5], x.last(1))
  127. assert_equal([3, 4, 5], x.last(3))
  128. assert_raise_with_message(ArgumentError, /0\.\.1/) {x.last(1, 2)}
  129. assert_equal(1, x.shift)
  130. assert_equal([2, 3, 4], x.shift(3))
  131. assert_equal([5], x)
  132. assert_raise_with_message(ArgumentError, /0\.\.1/) {x.first(1, 2)}
  133. assert_equal([2, 3, 4, 5], x.unshift(2, 3, 4))
  134. assert_equal([1, 2, 3, 4, 5], x.unshift(1))
  135. assert_equal([1, 2, 3, 4, 5], x)
  136. assert_equal(5, x.pop)
  137. assert_equal([3, 4], x.pop(2))
  138. assert_equal([1, 2], x)
  139. assert_raise_with_message(ArgumentError, /0\.\.1/) {x.pop(1, 2)}
  140. assert_equal([1, 2, 3, 4], x.push(3, 4))
  141. assert_equal([1, 2, 3, 4, 5], x.push(5))
  142. assert_equal([1, 2, 3, 4, 5], x)
  143. end
  144. def test_find_all_0
  145. assert_respond_to([], :find_all)
  146. assert_respond_to([], :select) # Alias
  147. assert_respond_to([], :filter) # Alias
  148. assert_equal([], [].find_all{ |obj| obj == "foo"})
  149. x = ["foo", "bar", "baz", "baz", 1, 2, 3, 3, 4]
  150. assert_equal(["baz","baz"], x.find_all{ |obj| obj == "baz" })
  151. assert_equal([3,3], x.find_all{ |obj| obj == 3 })
  152. end
  153. def test_fill_0
  154. assert_equal([-1, -1, -1, -1, -1, -1], [0, 1, 2, 3, 4, 5].fill(-1))
  155. assert_equal([0, 1, 2, -1, -1, -1], [0, 1, 2, 3, 4, 5].fill(-1, 3))
  156. assert_equal([0, 1, 2, -1, -1, 5], [0, 1, 2, 3, 4, 5].fill(-1, 3, 2))
  157. assert_equal([0, 1, 2, -1, -1, -1, -1, -1], [0, 1, 2, 3, 4, 5].fill(-1, 3, 5))
  158. assert_equal([0, 1, -1, -1, 4, 5], [0, 1, 2, 3, 4, 5].fill(-1, 2, 2))
  159. assert_equal([0, 1, -1, -1, -1, -1, -1], [0, 1, 2, 3, 4, 5].fill(-1, 2, 5))
  160. assert_equal([0, 1, 2, 3, -1, 5], [0, 1, 2, 3, 4, 5].fill(-1, -2, 1))
  161. assert_equal([0, 1, 2, 3, -1, -1, -1], [0, 1, 2, 3, 4, 5].fill(-1, -2, 3))
  162. assert_equal([0, 1, 2, -1, -1, 5], [0, 1, 2, 3, 4, 5].fill(-1, 3..4))
  163. assert_equal([0, 1, 2, -1, 4, 5], [0, 1, 2, 3, 4, 5].fill(-1, 3...4))
  164. assert_equal([0, 1, -1, -1, -1, 5], [0, 1, 2, 3, 4, 5].fill(-1, 2..-2))
  165. assert_equal([0, 1, -1, -1, 4, 5], [0, 1, 2, 3, 4, 5].fill(-1, 2...-2))
  166. assert_equal([10, 11, 12, 13, 14, 15], [0, 1, 2, 3, 4, 5].fill{|i| i+10})
  167. assert_equal([0, 1, 2, 13, 14, 15], [0, 1, 2, 3, 4, 5].fill(3){|i| i+10})
  168. assert_equal([0, 1, 2, 13, 14, 5], [0, 1, 2, 3, 4, 5].fill(3, 2){|i| i+10})
  169. assert_equal([0, 1, 2, 13, 14, 15, 16, 17], [0, 1, 2, 3, 4, 5].fill(3, 5){|i| i+10})
  170. assert_equal([0, 1, 2, 13, 14, 5], [0, 1, 2, 3, 4, 5].fill(3..4){|i| i+10})
  171. assert_equal([0, 1, 2, 13, 4, 5], [0, 1, 2, 3, 4, 5].fill(3...4){|i| i+10})
  172. assert_equal([0, 1, 12, 13, 14, 5], [0, 1, 2, 3, 4, 5].fill(2..-2){|i| i+10})
  173. assert_equal([0, 1, 12, 13, 4, 5], [0, 1, 2, 3, 4, 5].fill(2...-2){|i| i+10})
  174. assert_equal([0, 1, 2, 13, 14, 15], [0, 1, 2, 3, 4, 5].fill(3..){|i| i+10})
  175. assert_equal([0, 1, 2, 13, 14, 15], [0, 1, 2, 3, 4, 5].fill(3...){|i| i+10})
  176. end
  177. # From rubicon
  178. def test_00_new
  179. a = @cls.new()
  180. assert_instance_of(@cls, a)
  181. assert_equal(0, a.length)
  182. assert_nil(a[0])
  183. end
  184. def test_01_square_brackets
  185. a = @cls[ 5, 4, 3, 2, 1 ]
  186. assert_instance_of(@cls, a)
  187. assert_equal(5, a.length)
  188. 5.times { |i| assert_equal(5-i, a[i]) }
  189. assert_nil(a[6])
  190. end
  191. def test_AND # '&'
  192. assert_equal(@cls[1, 3], @cls[ 1, 1, 3, 5 ] & @cls[ 1, 2, 3 ])
  193. assert_equal(@cls[], @cls[ 1, 1, 3, 5 ] & @cls[ ])
  194. assert_equal(@cls[], @cls[ ] & @cls[ 1, 2, 3 ])
  195. assert_equal(@cls[], @cls[ 1, 2, 3 ] & @cls[ 4, 5, 6 ])
  196. end
  197. def test_AND_big_array # '&'
  198. assert_equal(@cls[1, 3], @cls[ 1, 1, 3, 5 ]*64 & @cls[ 1, 2, 3 ]*64)
  199. assert_equal(@cls[], @cls[ 1, 1, 3, 5 ]*64 & @cls[ ])
  200. assert_equal(@cls[], @cls[ ] & @cls[ 1, 2, 3 ]*64)
  201. assert_equal(@cls[], @cls[ 1, 2, 3 ]*64 & @cls[ 4, 5, 6 ]*64)
  202. end
  203. def test_intersection
  204. assert_equal(@cls[1, 2], @cls[1, 2, 3].intersection(@cls[1, 2]))
  205. assert_equal(@cls[ ], @cls[1].intersection(@cls[ ]))
  206. assert_equal(@cls[ ], @cls[ ].intersection(@cls[1]))
  207. assert_equal(@cls[1], @cls[1, 2, 3].intersection(@cls[1, 2], @cls[1]))
  208. assert_equal(@cls[ ], @cls[1, 2, 3].intersection(@cls[1, 2], @cls[3]))
  209. assert_equal(@cls[ ], @cls[1, 2, 3].intersection(@cls[4, 5, 6]))
  210. end
  211. def test_intersection_big_array
  212. assert_equal(@cls[1, 2], (@cls[1, 2, 3] * 64).intersection(@cls[1, 2] * 64))
  213. assert_equal(@cls[ ], (@cls[1] * 64).intersection(@cls[ ]))
  214. assert_equal(@cls[ ], @cls[ ].intersection(@cls[1] * 64))
  215. assert_equal(@cls[1], (@cls[1, 2, 3] * 64).intersection((@cls[1, 2] * 64), (@cls[1] * 64)))
  216. assert_equal(@cls[ ], (@cls[1, 2, 3] * 64).intersection(@cls[4, 5, 6] * 64))
  217. end
  218. def test_MUL # '*'
  219. assert_equal(@cls[], @cls[]*3)
  220. assert_equal(@cls[1, 1, 1], @cls[1]*3)
  221. assert_equal(@cls[1, 2, 1, 2, 1, 2], @cls[1, 2]*3)
  222. assert_equal(@cls[], @cls[1, 2, 3] * 0)
  223. assert_raise(ArgumentError) { @cls[1, 2]*(-3) }
  224. assert_equal('1-2-3-4-5', @cls[1, 2, 3, 4, 5] * '-')
  225. assert_equal('12345', @cls[1, 2, 3, 4, 5] * '')
  226. end
  227. def test_PLUS # '+'
  228. assert_equal(@cls[], @cls[] + @cls[])
  229. assert_equal(@cls[1], @cls[1] + @cls[])
  230. assert_equal(@cls[1], @cls[] + @cls[1])
  231. assert_equal(@cls[1, 1], @cls[1] + @cls[1])
  232. assert_equal(@cls['cat', 'dog', 1, 2, 3], %w(cat dog) + (1..3).to_a)
  233. end
  234. def test_MINUS # '-'
  235. assert_equal(@cls[], @cls[1] - @cls[1])
  236. assert_equal(@cls[1], @cls[1, 2, 3, 4, 5] - @cls[2, 3, 4, 5])
  237. assert_equal(@cls[1, 1, 1, 1], @cls[1, 2, 1, 3, 1, 4, 1, 5] - @cls[2, 3, 4, 5])
  238. assert_equal(@cls[1, 1], @cls[1, 2, 1] - @cls[2])
  239. assert_equal(@cls[1, 2, 3], @cls[1, 2, 3] - @cls[4, 5, 6])
  240. end
  241. def test_MINUS_big_array # '-'
  242. assert_equal(@cls[1]*64, @cls[1, 2, 3, 4, 5]*64 - @cls[2, 3, 4, 5]*64)
  243. assert_equal(@cls[1, 1, 1, 1]*64, @cls[1, 2, 1, 3, 1, 4, 1, 5]*64 - @cls[2, 3, 4, 5]*64)
  244. a = @cls[]
  245. 1000.times { a << 1 }
  246. assert_equal(1000, a.length)
  247. assert_equal(@cls[1] * 1000, a - @cls[2])
  248. end
  249. def test_difference
  250. assert_equal(@cls[], @cls[1].difference(@cls[1]))
  251. assert_equal(@cls[1], @cls[1, 2, 3, 4, 5].difference(@cls[2, 3, 4, 5]))
  252. assert_equal(@cls[1, 1], @cls[1, 2, 1].difference(@cls[2]))
  253. assert_equal(@cls[1, 1, 1, 1], @cls[1, 2, 1, 3, 1, 4, 1, 5].difference(@cls[2, 3, 4, 5]))
  254. assert_equal(@cls[], @cls[1, 2, 3, 4].difference(@cls[1], @cls[2], @cls[3], @cls[4]))
  255. a = [1]
  256. assert_equal(@cls[1], a.difference(@cls[2], @cls[2]))
  257. assert_equal(@cls[], a.difference(@cls[1]))
  258. assert_equal(@cls[1], a)
  259. end
  260. def test_difference_big_array
  261. assert_equal(@cls[1]*64, (@cls[1, 2, 3, 4, 5] * 64).difference(@cls[2, 3, 4] * 64, @cls[3, 5] * 64))
  262. assert_equal(@cls[1, 1, 1, 1]*64, (@cls[1, 2, 1, 3, 1, 4, 1, 5] * 64).difference(@cls[2, 3, 4, 5] * 64))
  263. a = @cls[1] * 1000
  264. assert_equal(@cls[1] * 1000, a.difference(@cls[2], @cls[2]))
  265. assert_equal(@cls[], a.difference(@cls[1]))
  266. assert_equal(@cls[1] * 1000, a)
  267. end
  268. def test_LSHIFT # '<<'
  269. a = @cls[]
  270. a << 1
  271. assert_equal(@cls[1], a)
  272. a << 2 << 3
  273. assert_equal(@cls[1, 2, 3], a)
  274. a << nil << 'cat'
  275. assert_equal(@cls[1, 2, 3, nil, 'cat'], a)
  276. a << a
  277. assert_equal(@cls[1, 2, 3, nil, 'cat', a], a)
  278. end
  279. def test_CMP # '<=>'
  280. assert_equal(0, @cls[] <=> @cls[])
  281. assert_equal(0, @cls[1] <=> @cls[1])
  282. assert_equal(0, @cls[1, 2, 3, 'cat'] <=> @cls[1, 2, 3, 'cat'])
  283. assert_equal(-1, @cls[] <=> @cls[1])
  284. assert_equal(1, @cls[1] <=> @cls[])
  285. assert_equal(-1, @cls[1, 2, 3] <=> @cls[1, 2, 3, 'cat'])
  286. assert_equal(1, @cls[1, 2, 3, 'cat'] <=> @cls[1, 2, 3])
  287. assert_equal(-1, @cls[1, 2, 3, 'cat'] <=> @cls[1, 2, 3, 'dog'])
  288. assert_equal(1, @cls[1, 2, 3, 'dog'] <=> @cls[1, 2, 3, 'cat'])
  289. end
  290. def test_EQUAL # '=='
  291. assert_operator(@cls[], :==, @cls[])
  292. assert_operator(@cls[1], :==, @cls[1])
  293. assert_operator(@cls[1, 1, 2, 2], :==, @cls[1, 1, 2, 2])
  294. assert_operator(@cls[1.0, 1.0, 2.0, 2.0], :==, @cls[1, 1, 2, 2])
  295. end
  296. def test_VERY_EQUAL # '==='
  297. assert_operator(@cls[], :===, @cls[])
  298. assert_operator(@cls[1], :===, @cls[1])
  299. assert_operator(@cls[1, 1, 2, 2], :===, @cls[1, 1, 2, 2])
  300. assert_operator(@cls[1.0, 1.0, 2.0, 2.0], :===, @cls[1, 1, 2, 2])
  301. end
  302. def test_AREF # '[]'
  303. a = @cls[*(1..100).to_a]
  304. assert_equal(1, a[0])
  305. assert_equal(100, a[99])
  306. assert_nil(a[100])
  307. assert_equal(100, a[-1])
  308. assert_equal(99, a[-2])
  309. assert_equal(1, a[-100])
  310. assert_nil(a[-101])
  311. assert_nil(a[-101,0])
  312. assert_nil(a[-101,1])
  313. assert_nil(a[-101,-1])
  314. assert_nil(a[10,-1])
  315. assert_equal(@cls[1], a[0,1])
  316. assert_equal(@cls[100], a[99,1])
  317. assert_equal(@cls[], a[100,1])
  318. assert_equal(@cls[100], a[99,100])
  319. assert_equal(@cls[100], a[-1,1])
  320. assert_equal(@cls[99], a[-2,1])
  321. assert_equal(@cls[], a[-100,0])
  322. assert_equal(@cls[1], a[-100,1])
  323. assert_equal(@cls[10, 11, 12], a[9, 3])
  324. assert_equal(@cls[10, 11, 12], a[-91, 3])
  325. assert_equal(@cls[1], a[0..0])
  326. assert_equal(@cls[100], a[99..99])
  327. assert_equal(@cls[], a[100..100])
  328. assert_equal(@cls[100], a[99..200])
  329. assert_equal(@cls[100], a[-1..-1])
  330. assert_equal(@cls[99], a[-2..-2])
  331. assert_equal(@cls[10, 11, 12], a[9..11])
  332. assert_equal(@cls[98, 99, 100], a[97..])
  333. assert_equal(@cls[1, 2, 3], a[..2])
  334. assert_equal(@cls[1, 2], a[...2])
  335. assert_equal(@cls[10, 11, 12], a[-91..-89])
  336. assert_equal(@cls[98, 99, 100], a[-3..])
  337. assert_equal(@cls[1, 2, 3], a[..-98])
  338. assert_equal(@cls[1, 2], a[...-98])
  339. assert_nil(a[10, -3])
  340. assert_equal [], a[10..7]
  341. assert_raise(TypeError) {a['cat']}
  342. end
  343. def test_ASET # '[]='
  344. a = @cls[*(0..99).to_a]
  345. assert_equal(0, a[0] = 0)
  346. assert_equal(@cls[0] + @cls[*(1..99).to_a], a)
  347. a = @cls[*(0..99).to_a]
  348. assert_equal(0, a[10,10] = 0)
  349. assert_equal(@cls[*(0..9).to_a] + @cls[0] + @cls[*(20..99).to_a], a)
  350. a = @cls[*(0..99).to_a]
  351. assert_equal(0, a[-1] = 0)
  352. assert_equal(@cls[*(0..98).to_a] + @cls[0], a)
  353. a = @cls[*(0..99).to_a]
  354. assert_equal(0, a[-10, 10] = 0)
  355. assert_equal(@cls[*(0..89).to_a] + @cls[0], a)
  356. a = @cls[*(0..99).to_a]
  357. assert_equal(0, a[0,1000] = 0)
  358. assert_equal(@cls[0] , a)
  359. a = @cls[*(0..99).to_a]
  360. assert_equal(0, a[10..19] = 0)
  361. assert_equal(@cls[*(0..9).to_a] + @cls[0] + @cls[*(20..99).to_a], a)
  362. b = @cls[*%w( a b c )]
  363. a = @cls[*(0..99).to_a]
  364. assert_equal(b, a[0,1] = b)
  365. assert_equal(b + @cls[*(1..99).to_a], a)
  366. a = @cls[*(0..99).to_a]
  367. assert_equal(b, a[10,10] = b)
  368. assert_equal(@cls[*(0..9).to_a] + b + @cls[*(20..99).to_a], a)
  369. a = @cls[*(0..99).to_a]
  370. assert_equal(b, a[-1, 1] = b)
  371. assert_equal(@cls[*(0..98).to_a] + b, a)
  372. a = @cls[*(0..99).to_a]
  373. assert_equal(b, a[-10, 10] = b)
  374. assert_equal(@cls[*(0..89).to_a] + b, a)
  375. a = @cls[*(0..99).to_a]
  376. assert_equal(b, a[0,1000] = b)
  377. assert_equal(b , a)
  378. a = @cls[*(0..99).to_a]
  379. assert_equal(b, a[10..19] = b)
  380. assert_equal(@cls[*(0..9).to_a] + b + @cls[*(20..99).to_a], a)
  381. a = @cls[*(0..99).to_a]
  382. assert_equal(nil, a[0,1] = nil)
  383. assert_equal(@cls[nil] + @cls[*(1..99).to_a], a)
  384. a = @cls[*(0..99).to_a]
  385. assert_equal(nil, a[10,10] = nil)
  386. assert_equal(@cls[*(0..9).to_a] + @cls[nil] + @cls[*(20..99).to_a], a)
  387. a = @cls[*(0..99).to_a]
  388. assert_equal(nil, a[-1, 1] = nil)
  389. assert_equal(@cls[*(0..98).to_a] + @cls[nil], a)
  390. a = @cls[*(0..99).to_a]
  391. assert_equal(nil, a[-10, 10] = nil)
  392. assert_equal(@cls[*(0..89).to_a] + @cls[nil], a)
  393. a = @cls[*(0..99).to_a]
  394. assert_equal(nil, a[0,1000] = nil)
  395. assert_equal(@cls[nil] , a)
  396. a = @cls[*(0..99).to_a]
  397. assert_equal(nil, a[10..19] = nil)
  398. assert_equal(@cls[*(0..9).to_a] + @cls[nil] + @cls[*(20..99).to_a], a)
  399. a = @cls[*(0..99).to_a]
  400. assert_equal(nil, a[10..] = nil)
  401. assert_equal(@cls[*(0..9).to_a] + @cls[nil], a)
  402. a = @cls[*(0..99).to_a]
  403. assert_equal(nil, a[..10] = nil)
  404. assert_equal(@cls[nil] + @cls[*(11..99).to_a], a)
  405. a = @cls[*(0..99).to_a]
  406. assert_equal(nil, a[...10] = nil)
  407. assert_equal(@cls[nil] + @cls[*(10..99).to_a], a)
  408. a = @cls[1, 2, 3]
  409. a[1, 0] = a
  410. assert_equal([1, 1, 2, 3, 2, 3], a)
  411. a = @cls[1, 2, 3]
  412. a[-1, 0] = a
  413. assert_equal([1, 2, 1, 2, 3, 3], a)
  414. a = @cls[]
  415. a[5,0] = [5]
  416. assert_equal([nil, nil, nil, nil, nil, 5], a)
  417. a = @cls[1]
  418. a[1,0] = [2]
  419. assert_equal([1, 2], a)
  420. a = @cls[1]
  421. a[1,1] = [2]
  422. assert_equal([1, 2], a)
  423. end
  424. def test_append
  425. a = @cls[1, 2, 3]
  426. assert_equal(@cls[1, 2, 3, 4, 5], a.append(4, 5))
  427. assert_equal(@cls[1, 2, 3, 4, 5, nil], a.append(nil))
  428. a.append
  429. assert_equal @cls[1, 2, 3, 4, 5, nil], a
  430. a.append 6, 7
  431. assert_equal @cls[1, 2, 3, 4, 5, nil, 6, 7], a
  432. end
  433. def test_assoc
  434. a1 = @cls[*%w( cat feline )]
  435. a2 = @cls[*%w( dog canine )]
  436. a3 = @cls[*%w( mule asinine )]
  437. a = @cls[ a1, a2, a3 ]
  438. assert_equal(a1, a.assoc('cat'))
  439. assert_equal(a3, a.assoc('mule'))
  440. assert_equal(nil, a.assoc('asinine'))
  441. assert_equal(nil, a.assoc('wombat'))
  442. assert_equal(nil, a.assoc(1..2))
  443. end
  444. def test_at
  445. a = @cls[*(0..99).to_a]
  446. assert_equal(0, a.at(0))
  447. assert_equal(10, a.at(10))
  448. assert_equal(99, a.at(99))
  449. assert_equal(nil, a.at(100))
  450. assert_equal(99, a.at(-1))
  451. assert_equal(0, a.at(-100))
  452. assert_equal(nil, a.at(-101))
  453. assert_raise(TypeError) { a.at('cat') }
  454. end
  455. def test_clear
  456. a = @cls[1, 2, 3]
  457. b = a.clear
  458. assert_equal(@cls[], a)
  459. assert_equal(@cls[], b)
  460. assert_equal(a.__id__, b.__id__)
  461. end
  462. def test_clone
  463. for frozen in [ false, true ]
  464. a = @cls[*(0..99).to_a]
  465. a.freeze if frozen
  466. b = a.clone
  467. assert_equal(a, b)
  468. assert_not_equal(a.__id__, b.__id__)
  469. assert_equal(a.frozen?, b.frozen?)
  470. end
  471. end
  472. def test_collect
  473. a = @cls[ 1, 'cat', 1..1 ]
  474. assert_equal([ Integer, String, Range], a.collect {|e| e.class} )
  475. assert_equal([ 99, 99, 99], a.collect { 99 } )
  476. assert_equal([], @cls[].collect { 99 })
  477. assert_kind_of Enumerator, @cls[1, 2, 3].collect
  478. assert_raise(ArgumentError) {
  479. assert_equal([[1, 2, 3]], [[1, 2, 3]].collect(&->(a, b, c) {[a, b, c]}))
  480. }
  481. end
  482. # also update map!
  483. def test_collect!
  484. a = @cls[ 1, 'cat', 1..1 ]
  485. assert_equal([ Integer, String, Range], a.collect! {|e| e.class} )
  486. assert_equal([ Integer, String, Range], a)
  487. a = @cls[ 1, 'cat', 1..1 ]
  488. assert_equal([ 99, 99, 99], a.collect! { 99 } )
  489. assert_equal([ 99, 99, 99], a)
  490. a = @cls[ ]
  491. assert_equal([], a.collect! { 99 })
  492. assert_equal([], a)
  493. end
  494. def test_compact
  495. a = @cls[ 1, nil, nil, 2, 3, nil, 4 ]
  496. assert_equal(@cls[1, 2, 3, 4], a.compact)
  497. a = @cls[ nil, 1, nil, 2, 3, nil, 4 ]
  498. assert_equal(@cls[1, 2, 3, 4], a.compact)
  499. a = @cls[ 1, nil, nil, 2, 3, nil, 4, nil ]
  500. assert_equal(@cls[1, 2, 3, 4], a.compact)
  501. a = @cls[ 1, 2, 3, 4 ]
  502. assert_equal(@cls[1, 2, 3, 4], a.compact)
  503. end
  504. def test_compact!
  505. a = @cls[ 1, nil, nil, 2, 3, nil, 4 ]
  506. assert_equal(@cls[1, 2, 3, 4], a.compact!)
  507. assert_equal(@cls[1, 2, 3, 4], a)
  508. a = @cls[ nil, 1, nil, 2, 3, nil, 4 ]
  509. assert_equal(@cls[1, 2, 3, 4], a.compact!)
  510. assert_equal(@cls[1, 2, 3, 4], a)
  511. a = @cls[ 1, nil, nil, 2, 3, nil, 4, nil ]
  512. assert_equal(@cls[1, 2, 3, 4], a.compact!)
  513. assert_equal(@cls[1, 2, 3, 4], a)
  514. a = @cls[ 1, 2, 3, 4 ]
  515. assert_equal(nil, a.compact!)
  516. assert_equal(@cls[1, 2, 3, 4], a)
  517. end
  518. def test_concat
  519. assert_equal(@cls[1, 2, 3, 4], @cls[1, 2].concat(@cls[3, 4]))
  520. assert_equal(@cls[1, 2, 3, 4], @cls[].concat(@cls[1, 2, 3, 4]))
  521. assert_equal(@cls[1, 2, 3, 4], @cls[1].concat(@cls[2, 3], [4]))
  522. assert_equal(@cls[1, 2, 3, 4], @cls[1, 2, 3, 4].concat(@cls[]))
  523. assert_equal(@cls[1, 2, 3, 4], @cls[1, 2, 3, 4].concat())
  524. assert_equal(@cls[], @cls[].concat(@cls[]))
  525. assert_equal(@cls[@cls[1, 2], @cls[3, 4]], @cls[@cls[1, 2]].concat(@cls[@cls[3, 4]]))
  526. a = @cls[1, 2, 3]
  527. a.concat(a)
  528. assert_equal([1, 2, 3, 1, 2, 3], a)
  529. b = @cls[4, 5]
  530. b.concat(b, b)
  531. assert_equal([4, 5, 4, 5, 4, 5], b)
  532. assert_raise(TypeError) { [0].concat(:foo) }
  533. assert_raise(FrozenError) { [0].freeze.concat(:foo) }
  534. end
  535. def test_count
  536. a = @cls[1, 2, 3, 1, 2]
  537. assert_equal(5, a.count)
  538. assert_equal(2, a.count(1))
  539. assert_equal(3, a.count {|x| x % 2 == 1 })
  540. assert_equal(2, a.count(1) {|x| x % 2 == 1 })
  541. assert_raise(ArgumentError) { a.count(0, 1) }
  542. bug8654 = '[ruby-core:56072]'
  543. assert_in_out_err [], <<-EOS, ["0"], [], bug8654
  544. a1 = []
  545. a2 = Array.new(100) { |i| i }
  546. a2.count do |i|
  547. p i
  548. a2.replace(a1) if i == 0
  549. end
  550. EOS
  551. assert_in_out_err [], <<-EOS, ["[]", "0"], [], bug8654
  552. ARY = Array.new(100) { |i| i }
  553. class Integer
  554. alias old_equal ==
  555. def == other
  556. ARY.replace([]) if self.equal?(0)
  557. p ARY
  558. self.equal?(other)
  559. end
  560. end
  561. p ARY.count(42)
  562. EOS
  563. end
  564. def test_delete
  565. a = @cls[*('cab'..'cat').to_a]
  566. assert_equal('cap', a.delete('cap'))
  567. assert_equal(@cls[*('cab'..'cao').to_a] + @cls[*('caq'..'cat').to_a], a)
  568. a = @cls[*('cab'..'cat').to_a]
  569. assert_equal('cab', a.delete('cab'))
  570. assert_equal(@cls[*('cac'..'cat').to_a], a)
  571. a = @cls[*('cab'..'cat').to_a]
  572. assert_equal('cat', a.delete('cat'))
  573. assert_equal(@cls[*('cab'..'cas').to_a], a)
  574. a = @cls[*('cab'..'cat').to_a]
  575. assert_equal(nil, a.delete('cup'))
  576. assert_equal(@cls[*('cab'..'cat').to_a], a)
  577. a = @cls[*('cab'..'cat').to_a]
  578. assert_equal(99, a.delete('cup') { 99 } )
  579. assert_equal(@cls[*('cab'..'cat').to_a], a)
  580. o = Object.new
  581. def o.==(other); true; end
  582. o2 = Object.new
  583. def o2.==(other); true; end
  584. a = @cls[1, o, o2, 2]
  585. assert_equal(o2, a.delete(42))
  586. assert_equal([1, 2], a)
  587. end
  588. def test_delete_at
  589. a = @cls[*(1..5).to_a]
  590. assert_equal(3, a.delete_at(2))
  591. assert_equal(@cls[1, 2, 4, 5], a)
  592. a = @cls[*(1..5).to_a]
  593. assert_equal(4, a.delete_at(-2))
  594. assert_equal(@cls[1, 2, 3, 5], a)
  595. a = @cls[*(1..5).to_a]
  596. assert_equal(nil, a.delete_at(5))
  597. assert_equal(@cls[1, 2, 3, 4, 5], a)
  598. a = @cls[*(1..5).to_a]
  599. assert_equal(nil, a.delete_at(-6))
  600. assert_equal(@cls[1, 2, 3, 4, 5], a)
  601. end
  602. # also reject!
  603. def test_delete_if
  604. a = @cls[ 1, 2, 3, 4, 5 ]
  605. assert_equal(a, a.delete_if { false })
  606. assert_equal(@cls[1, 2, 3, 4, 5], a)
  607. a = @cls[ 1, 2, 3, 4, 5 ]
  608. assert_equal(a, a.delete_if { true })
  609. assert_equal(@cls[], a)
  610. a = @cls[ 1, 2, 3, 4, 5 ]
  611. assert_equal(a, a.delete_if { |i| i > 3 })
  612. assert_equal(@cls[1, 2, 3], a)
  613. bug2545 = '[ruby-core:27366]'
  614. a = @cls[ 5, 6, 7, 8, 9, 10 ]
  615. assert_equal(9, a.delete_if {|i| break i if i > 8; i < 7})
  616. assert_equal(@cls[7, 8, 9, 10], a, bug2545)
  617. end
  618. def test_dup
  619. for frozen in [ false, true ]
  620. a = @cls[*(0..99).to_a]
  621. a.freeze if frozen
  622. b = a.dup
  623. assert_equal(a, b)
  624. assert_not_equal(a.__id__, b.__id__)
  625. assert_equal(false, b.frozen?)
  626. end
  627. end
  628. def test_each
  629. a = @cls[*%w( ant bat cat dog )]
  630. i = 0
  631. a.each { |e|
  632. assert_equal(a[i], e)
  633. i += 1
  634. }
  635. assert_equal(4, i)
  636. a = @cls[]
  637. i = 0
  638. a.each { |e|
  639. assert(false, "Never get here")
  640. i += 1
  641. }
  642. assert_equal(0, i)
  643. assert_equal(a, a.each {})
  644. end
  645. def test_each_index
  646. a = @cls[*%w( ant bat cat dog )]
  647. i = 0
  648. a.each_index { |ind|
  649. assert_equal(i, ind)
  650. i += 1
  651. }
  652. assert_equal(4, i)
  653. a = @cls[]
  654. i = 0
  655. a.each_index { |ind|
  656. assert(false, "Never get here")
  657. i += 1
  658. }
  659. assert_equal(0, i)
  660. assert_equal(a, a.each_index {})
  661. end
  662. def test_empty?
  663. assert_empty(@cls[])
  664. assert_not_empty(@cls[1])
  665. end
  666. def test_eql?
  667. assert_send([@cls[], :eql?, @cls[]])
  668. assert_send([@cls[1], :eql?, @cls[1]])
  669. assert_send([@cls[1, 1, 2, 2], :eql?, @cls[1, 1, 2, 2]])
  670. assert_not_send([@cls[1.0, 1.0, 2.0, 2.0], :eql?, @cls[1, 1, 2, 2]])
  671. end
  672. def test_fill
  673. assert_equal(@cls[], @cls[].fill(99))
  674. assert_equal(@cls[], @cls[].fill(99, 0))
  675. assert_equal(@cls[99], @cls[].fill(99, 0, 1))
  676. assert_equal(@cls[99], @cls[].fill(99, 0..0))
  677. assert_equal(@cls[99], @cls[1].fill(99))
  678. assert_equal(@cls[99], @cls[1].fill(99, 0))
  679. assert_equal(@cls[99], @cls[1].fill(99, 0, 1))
  680. assert_equal(@cls[99], @cls[1].fill(99, 0..0))
  681. assert_equal(@cls[99, 99], @cls[1, 2].fill(99))
  682. assert_equal(@cls[99, 99], @cls[1, 2].fill(99, 0))
  683. assert_equal(@cls[99, 99], @cls[1, 2].fill(99, nil))
  684. assert_equal(@cls[1, 99], @cls[1, 2].fill(99, 1, nil))
  685. assert_equal(@cls[99, 2], @cls[1, 2].fill(99, 0, 1))
  686. assert_equal(@cls[99, 2], @cls[1, 2].fill(99, 0..0))
  687. end
  688. def test_first
  689. assert_equal(3, @cls[3, 4, 5].first)
  690. assert_equal(nil, @cls[].first)
  691. end
  692. def test_flatten
  693. a1 = @cls[ 1, 2, 3]
  694. a2 = @cls[ 5, 6 ]
  695. a3 = @cls[ 4, a2 ]
  696. a4 = @cls[ a1, a3 ]
  697. assert_equal(@cls[1, 2, 3, 4, 5, 6], a4.flatten)
  698. assert_equal(@cls[ a1, a3], a4)
  699. a5 = @cls[ a1, @cls[], a3 ]
  700. assert_equal(@cls[1, 2, 3, 4, 5, 6], a5.flatten)
  701. assert_equal(@cls[1, 2, 3, 4, [5, 6]], a5.flatten(1))
  702. assert_equal(@cls[], @cls[].flatten)
  703. assert_equal(@cls[],
  704. @cls[@cls[@cls[@cls[],@cls[]],@cls[@cls[]],@cls[]],@cls[@cls[@cls[]]]].flatten)
  705. end
  706. def test_flatten_wrong_argument
  707. assert_raise(TypeError, "[ruby-dev:31197]") { [[]].flatten("") }
  708. end
  709. def test_flatten_level0
  710. a8 = @cls[[1, 2], 3]
  711. a9 = a8.flatten(0)
  712. assert_equal(a8, a9)
  713. assert_not_same(a8, a9)
  714. end
  715. def test_flatten_splat
  716. bug10748 = '[ruby-core:67637] [Bug #10748]'
  717. o = Object.new
  718. o.singleton_class.class_eval do
  719. define_method(:to_ary) do
  720. raise bug10748
  721. end
  722. end
  723. a = @cls[@cls[o]]
  724. assert_raise_with_message(RuntimeError, bug10748) {a.flatten}
  725. assert_nothing_raised(RuntimeError, bug10748) {a.flatten(1)}
  726. end
  727. def test_flattern_singleton_class
  728. bug12738 = '[ruby-dev:49781] [Bug #12738]'
  729. a = [[0]]
  730. class << a
  731. def m; end
  732. end
  733. assert_raise(NoMethodError, bug12738) { a.flatten.m }
  734. end
  735. def test_flatten!
  736. a1 = @cls[ 1, 2, 3]
  737. a2 = @cls[ 5, 6 ]
  738. a3 = @cls[ 4, a2 ]
  739. a4 = @cls[ a1, a3 ]
  740. assert_equal(@cls[1, 2, 3, 4, 5, 6], a4.flatten!)
  741. assert_equal(@cls[1, 2, 3, 4, 5, 6], a4)
  742. a5 = @cls[ a1, @cls[], a3 ]
  743. assert_equal(@cls[1, 2, 3, 4, 5, 6], a5.flatten!)
  744. assert_nil(a5.flatten!(0), '[ruby-core:23382]')
  745. assert_equal(@cls[1, 2, 3, 4, 5, 6], a5)
  746. end
  747. def test_flatten_empty!
  748. assert_nil(@cls[].flatten!)
  749. assert_equal(@cls[],
  750. @cls[@cls[@cls[@cls[],@cls[]],@cls[@cls[]],@cls[]],@cls[@cls[@cls[]]]].flatten!)
  751. end
  752. def test_flatten_level0!
  753. assert_nil(@cls[].flatten!(0), '[ruby-core:23382]')
  754. end
  755. def test_flatten_splat!
  756. bug10748 = '[ruby-core:67637] [Bug #10748]'
  757. o = Object.new
  758. o.singleton_class.class_eval do
  759. define_method(:to_ary) do
  760. raise bug10748
  761. end
  762. end
  763. a = @cls[@cls[o]]
  764. assert_raise_with_message(RuntimeError, bug10748) {a.flatten!}
  765. assert_nothing_raised(RuntimeError, bug10748) {a.flatten!(1)}
  766. end
  767. def test_flattern_singleton_class!
  768. bug12738 = '[ruby-dev:49781] [Bug #12738]'
  769. a = [[0]]
  770. class << a
  771. def m; end
  772. end
  773. assert_nothing_raised(NameError, bug12738) { a.flatten!.m }
  774. end
  775. def test_flatten_with_callcc
  776. need_continuation
  777. o = Object.new
  778. def o.to_ary() callcc {|k| @cont = k; [1,2,3]} end
  779. begin
  780. assert_equal([10, 20, 1, 2, 3, 30, 1, 2, 3, 40], [10, 20, o, 30, o, 40].flatten)
  781. rescue => e
  782. else
  783. o.instance_eval {@cont}.call
  784. end
  785. assert_instance_of(RuntimeError, e, '[ruby-dev:34798]')
  786. assert_match(/reentered/, e.message, '[ruby-dev:34798]')
  787. end
  788. def test_flatten_respond_to_missing
  789. bug11465 = '[ruby-core:70460] [Bug #11465]'
  790. obj = Class.new do
  791. def respond_to_missing?(method, stuff)
  792. return false if method == :to_ary
  793. super
  794. end
  795. def method_missing(*args)
  796. super
  797. end
  798. end.new
  799. ex = nil
  800. trace = TracePoint.new(:raise) do |tp|
  801. ex = tp.raised_exception
  802. end
  803. trace.enable {[obj].flatten}
  804. assert_nil(ex, bug11465)
  805. end
  806. def test_permutation_with_callcc
  807. need_continuation
  808. n = 1000
  809. cont = nil
  810. ary = [1,2,3]
  811. begin
  812. ary.permutation {
  813. callcc {|k| cont = k} unless cont
  814. }
  815. rescue => e
  816. end
  817. n -= 1
  818. cont.call if 0 < n
  819. assert_instance_of(RuntimeError, e)
  820. assert_match(/reentered/, e.message)
  821. end
  822. def test_product_with_callcc
  823. need_continuation
  824. n = 1000
  825. cont = nil
  826. ary = [1,2,3]
  827. begin
  828. ary.product {
  829. callcc {|k| cont = k} unless cont
  830. }
  831. rescue => e
  832. end
  833. n -= 1
  834. cont.call if 0 < n
  835. assert_instance_of(RuntimeError, e)
  836. assert_match(/reentered/, e.message)
  837. end
  838. def test_combination_with_callcc
  839. need_continuation
  840. n = 1000
  841. cont = nil
  842. ary = [1,2,3]
  843. begin
  844. ary.combination(2) {
  845. callcc {|k| cont = k} unless cont
  846. }
  847. rescue => e
  848. end
  849. n -= 1
  850. cont.call if 0 < n
  851. assert_instance_of(RuntimeError, e)
  852. assert_match(/reentered/, e.message)
  853. end
  854. def test_repeated_permutation_with_callcc
  855. need_continuation
  856. n = 1000
  857. cont = nil
  858. ary = [1,2,3]
  859. begin
  860. ary.repeated_permutation(2) {
  861. callcc {|k| cont = k} unless cont
  862. }
  863. rescue => e
  864. end
  865. n -= 1
  866. cont.call if 0 < n
  867. assert_instance_of(RuntimeError, e)
  868. assert_match(/reentered/, e.message)
  869. end
  870. def test_repeated_combination_with_callcc
  871. need_continuation
  872. n = 1000
  873. cont = nil
  874. ary = [1,2,3]
  875. begin
  876. ary.repeated_combination(2) {
  877. callcc {|k| cont = k} unless cont
  878. }
  879. rescue => e
  880. end
  881. n -= 1
  882. cont.call if 0 < n
  883. assert_instance_of(RuntimeError, e)
  884. assert_match(/reentered/, e.message)
  885. end
  886. def test_hash
  887. a1 = @cls[ 'cat', 'dog' ]
  888. a2 = @cls[ 'cat', 'dog' ]
  889. a3 = @cls[ 'dog', 'cat' ]
  890. assert_equal(a1.hash, a2.hash)
  891. assert_not_equal(a1.hash, a3.hash)
  892. bug9231 = '[ruby-core:58993] [Bug #9231]'
  893. assert_not_equal(false.hash, @cls[].hash, bug9231)
  894. end
  895. def test_include?
  896. a = @cls[ 'cat', 99, /a/, @cls[ 1, 2, 3] ]
  897. assert_include(a, 'cat')
  898. assert_include(a, 99)
  899. assert_include(a, /a/)
  900. assert_include(a, [1,2,3])
  901. assert_not_include(a, 'ca')
  902. assert_not_include(a, [1,2])
  903. end
  904. def test_index
  905. a = @cls[ 'cat', 99, /a/, 99, @cls[ 1, 2, 3] ]
  906. assert_equal(0, a.index('cat'))
  907. assert_equal(1, a.index(99))
  908. assert_equal(4, a.index([1,2,3]))
  909. assert_nil(a.index('ca'))
  910. assert_nil(a.index([1,2]))
  911. assert_equal(1, a.index(99) {|x| x == 'cat' })
  912. end
  913. def test_values_at
  914. a = @cls[*('a'..'j').to_a]
  915. assert_equal(@cls['a', 'c', 'e'], a.values_at(0, 2, 4))
  916. assert_equal(@cls['j', 'h', 'f'], a.values_at(-1, -3, -5))
  917. assert_equal(@cls['h', nil, 'a'], a.values_at(-3, 99, 0))
  918. end
  919. def test_join
  920. $, = ""
  921. a = @cls[]
  922. assert_equal("", a.join)
  923. assert_equal("", a.join(','))
  924. assert_equal(Encoding::US_ASCII, a.join.encoding)
  925. $, = ""
  926. a = @cls[1, 2]
  927. assert_equal("12", a.join)
  928. assert_equal("12", a.join(nil))
  929. assert_equal("1,2", a.join(','))
  930. $, = ""
  931. a = @cls[1, 2, 3]
  932. assert_equal("123", a.join)
  933. assert_equal("123", a.join(nil))
  934. assert_equal("1,2,3", a.join(','))
  935. $, = ":"
  936. a = @cls[1, 2, 3]
  937. assert_equal("1:2:3", a.join)
  938. assert_equal("1:2:3", a.join(nil))
  939. assert_equal("1,2,3", a.join(','))
  940. $, = ""
  941. e = ''.force_encoding('EUC-JP')
  942. u = ''.force_encoding('UTF-8')
  943. assert_equal(Encoding::US_ASCII, [[]].join.encoding)
  944. assert_equal(Encoding::US_ASCII, [1, [u]].join.encoding)
  945. assert_equal(Encoding::UTF_8, [u, [e]].join.encoding)
  946. assert_equal(Encoding::UTF_8, [u, [1]].join.encoding)
  947. assert_equal(Encoding::UTF_8, [Struct.new(:to_str).new(u)].join.encoding)
  948. bug5379 = '[ruby-core:39776]'
  949. assert_equal(Encoding::US_ASCII, [[], u, nil].join.encoding, bug5379)
  950. assert_equal(Encoding::UTF_8, [[], "\u3042", nil].join.encoding, bug5379)
  951. ensure
  952. $, = nil
  953. end
  954. def test_last
  955. assert_equal(nil, @cls[].last)
  956. assert_equal(1, @cls[1].last)
  957. assert_equal(99, @cls[*(3..99).to_a].last)
  958. end
  959. def test_length
  960. assert_equal(0, @cls[].length)
  961. assert_equal(1, @cls[1].length)
  962. assert_equal(2, @cls[1, nil].length)
  963. assert_equal(2, @cls[nil, 1].length)
  964. assert_equal(234, @cls[*(0..233).to_a].length)
  965. end
  966. # also update collect!
  967. def test_map!
  968. a = @cls[ 1, 'cat', 1..1 ]
  969. assert_equal(@cls[ Integer, String, Range], a.map! {|e| e.class} )
  970. assert_equal(@cls[ Integer, String, Range], a)
  971. a = @cls[ 1, 'cat', 1..1 ]
  972. assert_equal(@cls[ 99, 99, 99], a.map! { 99 } )
  973. assert_equal(@cls[ 99, 99, 99], a)
  974. a = @cls[ ]
  975. assert_equal(@cls[], a.map! { 99 })
  976. assert_equal(@cls[], a)
  977. end
  978. def test_pack
  979. a = @cls[*%w( cat wombat x yy)]
  980. assert_equal("catwomx yy ", a.pack("A3A3A3A3"))
  981. assert_equal("cat", a.pack("A*"))
  982. assert_equal("cwx yy ", a.pack("A3@1A3@2A3A3"))
  983. assert_equal("catwomx\000\000yy\000", a.pack("a3a3a3a3"))
  984. assert_equal("cat", a.pack("a*"))
  985. assert_equal("ca", a.pack("a2"))
  986. assert_equal("cat\000\000", a.pack("a5"))
  987. assert_equal("\x61", @cls["01100001"].pack("B8"))
  988. assert_equal("\x61", @cls["01100001"].pack("B*"))
  989. assert_equal("\x61", @cls["0110000100110111"].pack("B8"))
  990. assert_equal("\x61\x37", @cls["0110000100110111"].pack("B16"))
  991. assert_equal("\x61\x37", @cls["01100001", "00110111"].pack("B8B8"))
  992. assert_equal("\x60", @cls["01100001"].pack("B4"))
  993. assert_equal("\x40", @cls["01100001"].pack("B2"))
  994. assert_equal("\x86", @cls["01100001"].pack("b8"))
  995. assert_equal("\x86", @cls["01100001"].pack("b*"))
  996. assert_equal("\x86", @cls["0110000100110111"].pack("b8"))
  997. assert_equal("\x86\xec", @cls["0110000100110111"].pack("b16"))
  998. assert_equal("\x86\xec", @cls["01100001", "00110111"].pack("b8b8"))
  999. assert_equal("\x06", @cls["01100001"].pack("b4"))
  1000. assert_equal("\x02", @cls["01100001"].pack("b2"))
  1001. assert_equal("ABC", @cls[ 65, 66, 67 ].pack("C3"))
  1002. assert_equal("\377BC", @cls[ -1, 66, 67 ].pack("C*"))
  1003. assert_equal("ABC", @cls[ 65, 66, 67 ].pack("c3"))
  1004. assert_equal("\377BC", @cls[ -1, 66, 67 ].pack("c*"))
  1005. assert_equal("AB\n\x10", @cls["4142", "0a", "12"].pack("H4H2H1"))
  1006. assert_equal("AB\n\x02", @cls["1424", "a0", "21"].pack("h4h2h1"))
  1007. assert_equal("abc=02def=\ncat=\n=01=\n",
  1008. @cls["abc\002def", "cat", "\001"].pack("M9M3M4"))
  1009. assert_equal("aGVsbG8K\n", @cls["hello\n"].pack("m"))
  1010. assert_equal(",:&5L;&\\*:&5L;&\\*\n", @cls["hello\nhello\n"].pack("u"))
  1011. assert_equal("\u{a9 42 2260}", @cls[0xa9, 0x42, 0x2260].pack("U*"))
  1012. format = "c2x5CCxsdils_l_a6";
  1013. # Need the expression in here to force ary[5] to be numeric. This avoids
  1014. # test2 failing because ary2 goes str->numeric->str and ary does not.
  1015. ary = [1, -100, 127, 128, 32767, 987.654321098/100.0,
  1016. 12345, 123456, -32767, -123456, "abcdef"]
  1017. x = ary.pack(format)
  1018. ary2 = x.unpack(format)
  1019. assert_equal(ary.length, ary2.length)
  1020. assert_equal(ary.join(':'), ary2.join(':'))
  1021. assert_not_nil(x =~ /def/)
  1022. =begin
  1023. skipping "Not tested:
  1024. D,d & double-precision float, native format\\
  1025. E & double-precision float, little-endian byte order\\
  1026. e & single-precision float, little-endian byte order\\
  1027. F,f & single-precision float, native format\\
  1028. G & double-precision float, network (big-endian) byte order\\
  1029. g & single-precision float, network (big-endian) byte order\\
  1030. I & unsigned integer\\
  1031. i & integer\\
  1032. L & unsigned long\\
  1033. l & long\\
  1034. N & long, network (big-endian) byte order\\
  1035. n & short, network (big-endian) byte-order\\
  1036. P & pointer to a structure (fixed-length string)\\
  1037. p & pointer to a null-terminated string\\
  1038. S & unsigned short\\
  1039. s & short\\
  1040. V & long, little-endian byte order\\
  1041. v & short, little-endian byte order\\
  1042. X & back up a byte\\
  1043. x & null byte\\
  1044. Z & ASCII string (null padded, count is width)\\
  1045. "
  1046. =end
  1047. end
  1048. def test_pop
  1049. a = @cls[ 'cat', 'dog' ]
  1050. assert_equal('dog', a.pop)
  1051. assert_equal(@cls['cat'], a)
  1052. assert_equal('cat', a.pop)
  1053. assert_equal(@cls[], a)
  1054. assert_nil(a.pop)
  1055. assert_equal(@cls[], a)
  1056. end
  1057. def test_prepend
  1058. a = @cls[]
  1059. assert_equal(@cls['cat'], a.prepend('cat'))
  1060. assert_equal(@cls['dog', 'cat'], a.prepend('dog'))
  1061. assert_equal(@cls[nil, 'dog', 'cat'], a.prepend(nil))
  1062. assert_equal(@cls[@cls[1,2], nil, 'dog', 'cat'], a.prepend(@cls[1, 2]))
  1063. end
  1064. def test_push
  1065. a = @cls[1, 2, 3]
  1066. assert_equal(@cls[1, 2, 3, 4, 5], a.push(4, 5))
  1067. assert_equal(@cls[1, 2, 3, 4, 5, nil], a.push(nil))
  1068. a.push
  1069. assert_equal @cls[1, 2, 3, 4, 5, nil], a
  1070. a.push 6, 7
  1071. assert_equal @cls[1, 2, 3, 4, 5, nil, 6, 7], a
  1072. end
  1073. def test_rassoc
  1074. a1 = @cls[*%w( cat feline )]
  1075. a2 = @cls[*%w( dog canine )]
  1076. a3 = @cls[*%w( mule asinine )]
  1077. a = @cls[ a1, a2, a3 ]
  1078. assert_equal(a1, a.rassoc('feline'))
  1079. assert_equal(a3, a.rassoc('asinine'))
  1080. assert_equal(nil, a.rassoc('dog'))
  1081. assert_equal(nil, a.rassoc('mule'))
  1082. assert_equal(nil, a.rassoc(1..2))
  1083. end
  1084. # also delete_if
  1085. def test_reject!
  1086. a = @cls[ 1, 2, 3, 4, 5 ]
  1087. assert_equal(nil, a.reject! { false })
  1088. assert_equal(@cls[1, 2, 3, 4, 5], a)
  1089. a = @cls[ 1, 2, 3, 4, 5 ]
  1090. assert_equal(a, a.reject! { true })
  1091. assert_equal(@cls[], a)
  1092. a = @cls[ 1, 2, 3, 4, 5 ]
  1093. assert_equal(a, a.reject! { |i| i > 3 })
  1094. assert_equal(@cls[1, 2, 3], a)
  1095. bug2545 = '[ruby-core:27366]'
  1096. a = @cls[ 5, 6, 7, 8, 9, 10 ]
  1097. assert_equal(9, a.reject! {|i| break i if i > 8; i < 7})
  1098. assert_equal(@cls[7, 8, 9, 10], a, bug2545)
  1099. end
  1100. def test_shared_array_reject!
  1101. c = []
  1102. b = [1, 2, 3, 4]
  1103. 3.times do
  1104. a = b.dup
  1105. c << a.dup
  1106. begin
  1107. a.reject! do |x|
  1108. case x
  1109. when 2 then true
  1110. when 3 then raise StandardError, 'Oops'
  1111. else false
  1112. end
  1113. end
  1114. rescue StandardError
  1115. end
  1116. c << a.dup
  1117. end
  1118. bug90781 = '[ruby-core:90781]'
  1119. assert_equal [[1, 2, 3, 4],
  1120. [1, 3, 4],
  1121. [1, 2, 3, 4],
  1122. [1, 3, 4],
  1123. [1, 2, 3, 4],
  1124. [1, 3, 4]], c, bug90781
  1125. end
  1126. def test_iseq_shared_array_reject!
  1127. c = []
  1128. 3.times do
  1129. a = [1, 2, 3, 4]
  1130. c << a.dup
  1131. begin
  1132. a.reject! do |x|
  1133. case x
  1134. when 2 then true
  1135. when 3 then raise StandardError, 'Oops'
  1136. else false
  1137. end
  1138. end
  1139. rescue StandardError
  1140. end
  1141. c << a.dup
  1142. end
  1143. bug90781 = '[ruby-core:90781]'
  1144. assert_equal [[1, 2, 3, 4],
  1145. [1, 3, 4],
  1146. [1, 2, 3, 4],
  1147. [1, 3, 4],
  1148. [1, 2, 3, 4],
  1149. [1, 3, 4]], c, bug90781
  1150. end
  1151. def test_replace
  1152. a = @cls[ 1, 2, 3]
  1153. a_id = a.__id__
  1154. assert_equal(@cls[4, 5, 6], a.replace(@cls[4, 5, 6]))
  1155. assert_equal(@cls[4, 5, 6], a)
  1156. assert_equal(a_id, a.__id__)
  1157. assert_equal(@cls[], a.replace(@cls[]))
  1158. fa = a.dup.freeze
  1159. assert_nothing_raised(RuntimeError) { a.replace(a) }
  1160. assert_raise(FrozenError) { fa.replace(fa) }
  1161. assert_raise(ArgumentError) { fa.replace() }
  1162. assert_raise(TypeError) { a.replace(42) }
  1163. assert_raise(FrozenError) { fa.replace(42) }
  1164. end
  1165. def test_reverse
  1166. a = @cls[*%w( dog cat bee ant )]
  1167. assert_equal(@cls[*%w(ant bee cat dog)], a.reverse)
  1168. assert_equal(@cls[*%w(dog cat bee ant)], a)
  1169. assert_equal(@cls[], @cls[].reverse)
  1170. end
  1171. def test_reverse!
  1172. a = @cls[*%w( dog cat bee ant )]
  1173. assert_equal(@cls[*%w(ant bee cat dog)], a.reverse!)
  1174. assert_equal(@cls[*%w(ant bee cat dog)], a)
  1175. assert_equal @cls[], @cls[].reverse!
  1176. end
  1177. def test_reverse_each
  1178. a = @cls[*%w( dog cat bee ant )]
  1179. i = a.length
  1180. a.reverse_each { |e|
  1181. i -= 1
  1182. assert_equal(a[i], e)
  1183. }
  1184. assert_equal(0, i)
  1185. a = @cls[]
  1186. i = 0
  1187. a.reverse_each { |e|
  1188. i += 1
  1189. assert(false, "Never get here")
  1190. }
  1191. assert_equal(0, i)
  1192. end
  1193. def test_rindex
  1194. a = @cls[ 'cat', 99, /a/, 99, [ 1, 2, 3] ]
  1195. assert_equal(0, a.rindex('cat'))
  1196. assert_equal(3, a.rindex(99))
  1197. assert_equal(4, a.rindex([1,2,3]))
  1198. assert_nil(a.rindex('ca'))
  1199. assert_nil(a.rindex([1,2]))
  1200. assert_equal(3, a.rindex(99) {|x| x == [1,2,3] })
  1201. bug15951 = "[Bug #15951]"
  1202. o2 = Object.new
  1203. def o2.==(other)
  1204. other.replace([]) if Array === other
  1205. false
  1206. end
  1207. a = Array.new(10)
  1208. a.fill(o2)
  1209. assert_nil(a.rindex(a), bug15951)
  1210. end
  1211. def test_shift
  1212. a = @cls[ 'cat', 'dog' ]
  1213. assert_equal('cat', a.shift)
  1214. assert_equal(@cls['dog'], a)
  1215. assert_equal('dog', a.shift)
  1216. assert_equal(@cls[], a)
  1217. assert_nil(a.shift)
  1218. assert_equal(@cls[], a)
  1219. end
  1220. def test_size
  1221. assert_equal(0, @cls[].size)
  1222. assert_equal(1, @cls[1].size)
  1223. assert_equal(100, @cls[*(0..99).to_a].size)
  1224. end
  1225. def test_slice
  1226. a = @cls[*(1..100).to_a]
  1227. assert_equal(1, a.slice(0))
  1228. assert_equal(100, a.slice(99))
  1229. assert_nil(a.slice(100))
  1230. assert_equal(100, a.slice(-1))
  1231. assert_equal(99, a.slice(-2))
  1232. assert_equal(1, a.slice(-100))
  1233. assert_nil(a.slice(-101))
  1234. assert_equal(@cls[1], a.slice(0,1))
  1235. assert_equal(@cls[100], a.slice(99,1))
  1236. assert_equal(@cls[], a.slice(100,1))
  1237. assert_equal(@cls[100], a.slice(99,100))
  1238. assert_equal(@cls[100], a.slice(-1,1))
  1239. assert_equal(@cls[99], a.slice(-2,1))
  1240. assert_equal(@cls[10, 11, 12], a.slice(9, 3))
  1241. assert_equal(@cls[10, 11, 12], a.slice(-91, 3))
  1242. assert_nil(a.slice(-101, 2))
  1243. assert_equal(@cls[1], a.slice(0..0))
  1244. assert_equal(@cls[100], a.slice(99..99))
  1245. assert_equal(@cls[], a.slice(100..100))
  1246. assert_equal(@cls[100], a.slice(99..200))
  1247. assert_equal(@cls[100], a.slice(-1..-1))
  1248. assert_equal(@cls[99], a.slice(-2..-2))
  1249. assert_equal(@cls[10, 11, 12], a.slice(9..11))
  1250. assert_equal(@cls[98, 99, 100], a.slice(97..))
  1251. assert_equal(@cls[10, 11, 12], a.slice(-91..-89))
  1252. assert_equal(@cls[10, 11, 12], a.slice(-91..-89))
  1253. assert_nil(a.slice(-101..-1))
  1254. assert_nil(a.slice(-101..))
  1255. assert_nil(a.slice(10, -3))
  1256. assert_equal @cls[], a.slice(10..7)
  1257. end
  1258. def test_slice!
  1259. a = @cls[1, 2, 3, 4, 5]
  1260. assert_equal(3, a.slice!(2))
  1261. assert_equal(@cls[1, 2, 4, 5], a)
  1262. a = @cls[1, 2, 3, 4, 5]
  1263. assert_equal(4, a.slice!(-2))
  1264. assert_equal(@cls[1, 2, 3, 5], a)
  1265. a = @cls[1, 2, 3, 4, 5]
  1266. assert_equal(@cls[3,4], a.slice!(2,2))
  1267. assert_equal(@cls[1, 2, 5], a)
  1268. a = @cls[1, 2, 3, 4, 5]
  1269. assert_equal(@cls[4,5], a.slice!(-2,2))
  1270. assert_equal(@cls[1, 2, 3], a)
  1271. a = @cls[1, 2, 3, 4, 5]
  1272. assert_equal(@cls[3,4], a.slice!(2..3))
  1273. assert_equal(@cls[1, 2, 5], a)
  1274. a = @cls[1, 2, 3, 4, 5]
  1275. assert_equal(nil, a.slice!(20))
  1276. assert_equal(@cls[1, 2, 3, 4, 5], a)
  1277. a = @cls[1, 2, 3, 4, 5]
  1278. assert_equal(nil, a.slice!(-6))
  1279. assert_equal(@cls[1, 2, 3, 4, 5], a)
  1280. a = @cls[1, 2, 3, 4, 5]
  1281. assert_equal(nil, a.slice!(-6..4))
  1282. assert_equal(@cls[1, 2, 3, 4, 5], a)
  1283. a = @cls[1, 2, 3, 4, 5]
  1284. assert_equal(nil, a.slice!(-6,2))
  1285. assert_equal(@cls[1, 2, 3, 4, 5], a)
  1286. assert_equal("[2, 3]", [1,2,3].slice!(1,10000).inspect, "moved from btest/knownbug")
  1287. assert_raise(ArgumentError) { @cls[1].slice! }
  1288. assert_raise(ArgumentError) { @cls[1].slice!(0, 0, 0) }
  1289. end
  1290. def test_sort
  1291. a = @cls[ 4, 1, 2, 3 ]
  1292. assert_equal(@cls[1, 2, 3, 4], a.sort)
  1293. assert_equal(@cls[4, 1, 2, 3], a)
  1294. assert_equal(@cls[4, 3, 2, 1], a.sort { |x, y| y <=> x} )
  1295. assert_equal(@cls[4, 1, 2, 3], a)
  1296. assert_equal(@cls[1, 2, 3, 4], a.sort { |x, y| (x - y) * (2**100) })
  1297. a.fill(1)
  1298. assert_equal(@cls[1, 1, 1, 1], a.sort)
  1299. assert_equal(@cls[], @cls[].sort)
  1300. end
  1301. def test_sort!
  1302. a = @cls[ 4, 1, 2, 3 ]
  1303. assert_equal(@cls[1, 2, 3, 4], a.sort!)
  1304. assert_equal(@cls[1, 2, 3, 4], a)
  1305. assert_equal(@cls[4, 3, 2, 1], a.sort! { |x, y| y <=> x} )
  1306. assert_equal(@cls[4, 3, 2, 1], a)
  1307. a.fill(1)
  1308. assert_equal(@cls[1, 1, 1, 1], a.sort!)
  1309. assert_equal(@cls[1], @cls[1].sort!)
  1310. assert_equal(@cls[], @cls[].sort!)
  1311. a = @cls[4, 3, 2, 1]
  1312. a.sort! {|m, n| a.replace([9, 8, 7, 6]); m <=> n }
  1313. assert_equal([1, 2, 3, 4], a)
  1314. a = @cls[4, 3, 2, 1]
  1315. a.sort! {|m, n| a.replace([9, 8, 7]); m <=> n }
  1316. assert_equal([1, 2, 3, 4], a)
  1317. end
  1318. def test_sort_with_callcc
  1319. need_continuation
  1320. n = 1000
  1321. cont = nil
  1322. ary = (1..100).to_a
  1323. begin
  1324. ary.sort! {|a,b|
  1325. callcc {|k| cont = k} unless cont
  1326. assert_equal(100, ary.size, '[ruby-core:16679]')
  1327. a <=> b
  1328. }
  1329. rescue => e
  1330. end
  1331. n -= 1
  1332. cont.call if 0 < n
  1333. assert_instance_of(RuntimeError, e, '[ruby-core:16679]')
  1334. assert_match(/reentered/, e.message, '[ruby-core:16679]')
  1335. end
  1336. def test_sort_with_replace
  1337. bug = '[ruby-core:34732]'
  1338. assert_separately([], "#{<<~"begin;"}\n#{<<~'end;'}", timeout: 30)
  1339. bug = "#{bug}"
  1340. begin;
  1341. xary = (1..100).to_a
  1342. 100.times do
  1343. ary = (1..100).to_a
  1344. ary.sort! {|a,b| ary.replace(xary); a <=> b}
  1345. GC.start
  1346. assert_equal(xary, ary, '[ruby-dev:34732]')
  1347. end
  1348. assert_nothing_raised(SystemStackError, bug) do
  1349. assert_equal(:ok, Array.new(100_000, nil).permutation {break :ok})
  1350. end
  1351. end;
  1352. end
  1353. def test_sort_bang_with_freeze
  1354. ary = []
  1355. o1 = Object.new
  1356. o1.singleton_class.class_eval {
  1357. define_method(:<=>) {|v|
  1358. ary.freeze
  1359. 1
  1360. }
  1361. }
  1362. o2 = o1.clone
  1363. ary << o1 << o2
  1364. orig = ary.dup
  1365. assert_raise(FrozenError, "frozen during comparison") {ary.sort!}
  1366. assert_equal(orig, ary, "must not be modified once frozen")
  1367. end
  1368. def test_short_heap_array_sort_bang_memory_leak
  1369. bug11332 = '[ruby-dev:49166] [Bug #11332]'
  1370. assert_no_memory_leak([], <<-PREP, <<-TEST, bug11332, limit: 1.3, timeout: 60)
  1371. def t; ary = [*1..5]; ary.pop(2); ary.sort!; end
  1372. 1.times {t}
  1373. PREP
  1374. 500000.times {t}
  1375. TEST
  1376. end
  1377. def test_to_a
  1378. a = @cls[ 1, 2, 3 ]
  1379. a_id = a.__id__
  1380. assert_equal(a, a.to_a)
  1381. assert_equal(a_id, a.to_a.__id__)
  1382. end
  1383. def test_to_ary
  1384. a = [ 1, 2, 3 ]
  1385. b = @cls[*a]
  1386. a_id = a.__id__
  1387. assert_equal(a, b.to_ary)
  1388. if (@cls == Array)
  1389. assert_equal(a_id, a.to_ary.__id__)
  1390. end
  1391. o = Object.new
  1392. def o.to_ary
  1393. [4, 5]
  1394. end
  1395. assert_equal([1, 2, 3, 4, 5], a.concat(o))
  1396. o = Object.new
  1397. def o.to_ary
  1398. foo_bar()
  1399. end
  1400. assert_raise_with_message(NoMethodError, /foo_bar/) {a.concat(o)}
  1401. end
  1402. def test_to_s
  1403. $, = ""
  1404. a = @cls[]
  1405. assert_equal("[]", a.to_s)
  1406. $, = ""
  1407. a = @cls[1, 2]
  1408. assert_equal("[1, 2]", a.to_s)
  1409. $, = ""
  1410. a = @cls[1, 2, 3]
  1411. assert_equal("[1, 2, 3]", a.to_s)
  1412. $, = ":"
  1413. a = @cls[1, 2, 3]
  1414. assert_equal("[1, 2, 3]", a.to_s)
  1415. ensure
  1416. $, = nil
  1417. end
  1418. StubToH = [
  1419. [:key, :value],
  1420. Object.new.tap do |kvp|
  1421. def kvp.to_ary
  1422. [:obtained, :via_to_ary]
  1423. end
  1424. end,
  1425. ]
  1426. def test_to_h
  1427. array = StubToH
  1428. assert_equal({key: :value, obtained: :via_to_ary}, array.to_h)
  1429. e = assert_raise(TypeError) {
  1430. [[:first_one, :ok], :not_ok].to_h
  1431. }
  1432. assert_equal "wrong element type Symbol at 1 (expected array)", e.message
  1433. array = [eval("class C\u{1f5ff}; self; end").new]
  1434. assert_raise_with_message(TypeError, /C\u{1f5ff}/) {array.to_h}
  1435. e = assert_raise(ArgumentError) {
  1436. [[:first_one, :ok], [1, 2], [:not_ok]].to_h
  1437. }
  1438. assert_equal "wrong array length at 2 (expected 2, was 1)", e.message
  1439. end
  1440. def test_to_h_block
  1441. array = StubToH
  1442. assert_equal({"key" => "value", "obtained" => "via_to_ary"},
  1443. array.to_h {|k, v| [k.to_s, v.to_s]})
  1444. assert_equal({first_one: :ok, not_ok: :ng},
  1445. [[:first_one, :ok], :not_ok].to_h {|k, v| [k, v || :ng]})
  1446. e = assert_raise(TypeError) {
  1447. [[:first_one, :ok], :not_ok].to_h {|k, v| v ? [k, v] : k}
  1448. }
  1449. assert_equal "wrong element type Symbol at 1 (expected array)", e.message
  1450. array = [1]
  1451. k = eval("class C\u{1f5ff}; self; end").new
  1452. assert_raise_with_message(TypeError, /C\u{1f5ff}/) {array.to_h {k}}
  1453. e = assert_raise(ArgumentError) {
  1454. [[:first_one, :ok], [1, 2], [:not_ok]].to_h {|kv| kv}
  1455. }
  1456. assert_equal "wrong array length at 2 (expected 2, was 1)", e.message
  1457. end
  1458. def test_min
  1459. assert_equal(1, [1, 2, 3, 1, 2].min)
  1460. assert_equal(3, [1, 2, 3, 1, 2].min {|a,b| b <=> a })
  1461. cond = ->((a, ia), (b, ib)) { (b <=> a).nonzero? or ia <=> ib }
  1462. assert_equal([3, 2], [1, 2, 3, 1, 2].each_with_index.min(&cond))
  1463. ary = %w(albatross dog horse)
  1464. assert_equal("albatross", ary.min)
  1465. assert_equal("dog", ary.min {|a,b| a.length <=> b.length })
  1466. assert_equal(1, [3,2,1].min)
  1467. assert_equal(%w[albatross dog], ary.min(2))
  1468. assert_equal(%w[dog horse],
  1469. ary.min(2) {|a,b| a.length <=> b.length })
  1470. assert_equal([13, 14], [20, 32, 32, 21, 30, 25, 29, 13, 14].min(2))
  1471. assert_equal([2, 4, 6, 7], [2, 4, 8, 6, 7].min(4))
  1472. class << (obj = Object.new)
  1473. def <=>(x) 1 <=> x end
  1474. def coerce(x) [x, 1] end
  1475. end
  1476. assert_same(obj, [obj, 1.0].min)
  1477. end
  1478. def test_max
  1479. assert_equal(3, [1, 2, 3, 1, 2].max)
  1480. assert_equal(1, [1, 2, 3, 1, 2].max {|a,b| b <=> a })
  1481. cond = ->((a, ia), (b, ib)) { (b <=> a).nonzero? or ia <=> ib }
  1482. assert_equal([1, 3], [1, 2, 3, 1, 2].each_with_index.max(&cond))
  1483. ary = %w(albatross dog horse)
  1484. assert_equal("horse", ary.max)
  1485. assert_equal("albatross", ary.max {|a,b| a.length <=> b.length })
  1486. assert_equal(1, [3,2,1].max{|a,b| b <=> a })
  1487. assert_equal(%w[horse dog], ary.max(2))
  1488. assert_equal(%w[albatross horse],
  1489. ary.max(2) {|a,b| a.length <=> b.length })
  1490. assert_equal([3, 2], [0, 0, 0, 0, 0, 0, 1, 3, 2].max(2))
  1491. class << (obj = Object.new)
  1492. def <=>(x) 1 <=> x end
  1493. def coerce(x) [x, 1] end
  1494. end
  1495. assert_same(obj, [obj, 1.0].max)
  1496. end
  1497. def test_minmax
  1498. assert_equal([1, 3], [1, 2, 3, 1, 2].minmax)
  1499. assert_equal([3, 1], [1, 2, 3, 1, 2].minmax {|a,b| b <=> a })
  1500. cond = ->((a, ia), (b, ib)) { (b <=> a).nonzero? or ia <=> ib }
  1501. assert_equal([[3, 2], [1, 3]], [1, 2, 3, 1, 2].each_with_index.minmax(&cond))
  1502. ary = %w(albatross dog horse)
  1503. assert_equal(["albatross", "horse"], ary.minmax)
  1504. assert_equal(["dog", "albatross"], ary.minmax {|a,b| a.length <=> b.length })
  1505. assert_equal([1, 3], [3,2,1].minmax)
  1506. class << (obj = Object.new)
  1507. def <=>(x) 1 <=> x end
  1508. def coerce(x) [x, 1] end
  1509. end
  1510. ary = [obj, 1.0].minmax
  1511. assert_same(obj, ary[0])
  1512. assert_equal(obj, ary[1])
  1513. end
  1514. def test_uniq
  1515. a = []
  1516. b = a.uniq
  1517. assert_equal([], a)
  1518. assert_equal([], b)
  1519. assert_not_same(a, b)
  1520. a = [1]
  1521. b = a.uniq
  1522. assert_equal([1], a)
  1523. assert_equal([1], b)
  1524. assert_not_same(a, b)
  1525. a = [1,1]
  1526. b = a.uniq
  1527. assert_equal([1,1], a)
  1528. assert_equal([1], b)
  1529. assert_not_same(a, b)
  1530. a = [1,2]
  1531. b = a.uniq
  1532. assert_equal([1,2], a)
  1533. assert_equal([1,2], b)
  1534. assert_not_same(a, b)
  1535. a = @cls[ 1, 2, 3, 2, 1, 2, 3, 4, nil ]
  1536. b = a.dup
  1537. assert_equal(@cls[1, 2, 3, 4, nil], a.uniq)
  1538. assert_equal(b, a)
  1539. c = @cls["a:def", "a:xyz", "b:abc", "b:xyz", "c:jkl"]
  1540. d = c.dup
  1541. assert_equal(@cls[ "a:def", "b:abc", "c:jkl" ], c.uniq {|s| s[/^\w+/]})
  1542. assert_equal(d, c)
  1543. assert_equal(@cls[1, 2, 3], @cls[1, 2, 3].uniq)
  1544. a = %w(a a)
  1545. b = a.uniq
  1546. assert_equal(%w(a a), a)
  1547. assert(a.none?(&:frozen?))
  1548. assert_equal(%w(a), b)
  1549. assert(b.none?(&:frozen?))
  1550. bug9340 = "[ruby-core:59457]"
  1551. ary = [bug9340, bug9340.dup, bug9340.dup]
  1552. assert_equal 1, ary.uniq.size
  1553. assert_same bug9340, ary.uniq[0]
  1554. sc = Class.new(@cls)
  1555. a = sc[]
  1556. b = a.dup
  1557. assert_instance_of(sc, a.uniq)
  1558. assert_equal(sc[], a.uniq)
  1559. assert_equal(b, a)
  1560. a = sc[1]
  1561. b = a.dup
  1562. assert_instance_of(sc, a.uniq)
  1563. assert_equal(sc[1], a.uniq)
  1564. assert_equal(b, a)
  1565. a = sc[1, 1]
  1566. b = a.dup
  1567. assert_instance_of(sc, a.uniq)
  1568. assert_equal(sc[1], a.uniq)
  1569. assert_equal(b, a)
  1570. a = sc[1, 1]
  1571. b = a.dup
  1572. assert_instance_of(sc, a.uniq{|x| x})
  1573. assert_equal(sc[1], a.uniq{|x| x})
  1574. assert_equal(b, a)
  1575. end
  1576. def test_uniq_with_block
  1577. a = []
  1578. b = a.uniq {|v| v.even? }
  1579. assert_equal([], a)
  1580. assert_equal([], b)
  1581. assert_not_same(a, b)
  1582. a = [1]
  1583. b = a.uniq {|v| v.even? }
  1584. assert_equal([1], a)
  1585. assert_equal([1], b)
  1586. assert_not_same(a, b)
  1587. a = [1,3]
  1588. b = a.uniq {|v| v.even? }
  1589. assert_equal([1,3], a)
  1590. assert_equal([1], b)
  1591. assert_not_same(a, b)
  1592. a = %w(a a)
  1593. b = a.uniq {|v| v }
  1594. assert_equal(%w(a a), a)
  1595. assert(a.none?(&:frozen?))
  1596. assert_equal(%w(a), b)
  1597. assert(b.none?(&:frozen?))
  1598. end
  1599. def test_uniq!
  1600. a = []
  1601. b = a.uniq!
  1602. assert_equal(nil, b)
  1603. a = [1]
  1604. b = a.uniq!
  1605. assert_equal(nil, b)
  1606. a = [1,1]
  1607. b = a.uniq!
  1608. assert_equal([1], a)
  1609. assert_equal([1], b)
  1610. assert_same(a, b)
  1611. a = [1,2]
  1612. b = a.uniq!
  1613. assert_equal([1,2], a)
  1614. assert_equal(nil, b)
  1615. a = @cls[ 1, 2, 3, 2, 1, 2, 3, 4, nil ]
  1616. assert_equal(@cls[1, 2, 3, 4, nil], a.uniq!)
  1617. assert_equal(@cls[1, 2, 3, 4, nil], a)
  1618. c = @cls["a:def", "a:xyz", "b:abc", "b:xyz", "c:jkl"]
  1619. assert_equal(@cls[ "a:def", "b:abc", "c:jkl" ], c.uniq! {|s| s[/^\w+/]})
  1620. assert_equal(@cls[ "a:def", "b:abc", "c:jkl" ], c)
  1621. c = @cls["a:def", "b:abc", "c:jkl"]
  1622. assert_equal(nil, c.uniq! {|s| s[/^\w+/]})
  1623. assert_equal(@cls[ "a:def", "b:abc", "c:jkl" ], c)
  1624. assert_nil(@cls[1, 2, 3].uniq!)
  1625. f = a.dup.freeze
  1626. assert_raise(ArgumentError) { a.uniq!(1) }
  1627. assert_raise(ArgumentError) { f.uniq!(1) }
  1628. assert_raise(FrozenError) { f.uniq! }
  1629. assert_nothing_raised do
  1630. a = [ {c: "b"}, {c: "r"}, {c: "w"}, {c: "g"}, {c: "g"} ]
  1631. a.sort_by!{|e| e[:c]}
  1632. a.uniq! {|e| e[:c]}
  1633. end
  1634. a = %w(a a)
  1635. b = a.uniq
  1636. assert_equal(%w(a a), a)
  1637. assert(a.none?(&:frozen?))
  1638. assert_equal(%w(a), b)
  1639. assert(b.none?(&:frozen?))
  1640. end
  1641. def test_uniq_bang_with_block
  1642. a = []
  1643. b = a.uniq! {|v| v.even? }
  1644. assert_equal(nil, b)
  1645. a = [1]
  1646. b = a.uniq! {|v| v.even? }
  1647. assert_equal(nil, b)
  1648. a = [1,3]
  1649. b = a.uniq! {|v| v.even? }
  1650. assert_equal([1], a)
  1651. assert_equal([1], b)
  1652. assert_same(a, b)
  1653. a = [1,2]
  1654. b = a.uniq! {|v| v.even? }
  1655. assert_equal([1,2], a)
  1656. assert_equal(nil, b)
  1657. a = %w(a a)
  1658. b = a.uniq! {|v| v }
  1659. assert_equal(%w(a), b)
  1660. assert_same(a, b)
  1661. assert b.none?(&:frozen?)
  1662. end
  1663. def test_uniq_bang_with_freeze
  1664. ary = [1,2]
  1665. orig = ary.dup
  1666. assert_raise(FrozenError, "frozen during comparison") {
  1667. ary.uniq! {|v| ary.freeze; 1}
  1668. }
  1669. assert_equal(orig, ary, "must not be modified once frozen")
  1670. end
  1671. def test_unshift
  1672. a = @cls[]
  1673. assert_equal(@cls['cat'], a.unshift('cat'))
  1674. assert_equal(@cls['dog', 'cat'], a.unshift('dog'))
  1675. assert_equal(@cls[nil, 'dog', 'cat'], a.unshift(nil))
  1676. assert_equal(@cls[@cls[1,2], nil, 'dog', 'cat'], a.unshift(@cls[1, 2]))
  1677. end
  1678. def test_unshift_frozen
  1679. bug15952 = '[Bug #15952]'
  1680. assert_raise(FrozenError, bug15952) do
  1681. a = [1] * 100
  1682. b = a[4..-1]
  1683. a.replace([1])
  1684. b.freeze
  1685. b.unshift("a")
  1686. end
  1687. end
  1688. def test_OR # '|'
  1689. assert_equal(@cls[], @cls[] | @cls[])
  1690. assert_equal(@cls[1], @cls[1] | @cls[])
  1691. assert_equal(@cls[1], @cls[] | @cls[1])
  1692. assert_equal(@cls[1], @cls[1] | @cls[1])
  1693. assert_equal(@cls[1,2], @cls[1] | @cls[2])
  1694. assert_equal(@cls[1,2], @cls[1, 1] | @cls[2, 2])
  1695. assert_equal(@cls[1,2], @cls[1, 2] | @cls[1, 2])
  1696. a = %w(a b c)
  1697. b = %w(a b c d e)
  1698. c = a | b
  1699. assert_equal(c, b)
  1700. assert_not_same(c, b)
  1701. assert_equal(%w(a b c), a)
  1702. assert_equal(%w(a b c d e), b)
  1703. assert(a.none?(&:frozen?))
  1704. assert(b.none?(&:frozen?))
  1705. assert(c.none?(&:frozen?))
  1706. end
  1707. def test_OR_in_order
  1708. obj1, obj2 = Class.new do
  1709. attr_reader :name
  1710. def initialize(name) @name = name; end
  1711. def inspect; "test_OR_in_order(#{@name})"; end
  1712. def hash; 0; end
  1713. def eql?(a) true; end
  1714. break [new("1"), new("2")]
  1715. end
  1716. assert_equal([obj1], [obj1]|[obj2])
  1717. end
  1718. def test_OR_big_in_order
  1719. obj1, obj2 = Class.new do
  1720. attr_reader :name
  1721. def initialize(name) @name = name; end
  1722. def inspect; "test_OR_in_order(#{@name})"; end
  1723. def hash; 0; end
  1724. def eql?(a) true; end
  1725. break [new("1"), new("2")]
  1726. end
  1727. assert_equal([obj1], [obj1]*64|[obj2]*64)
  1728. end
  1729. def test_OR_big_array # '|'
  1730. assert_equal(@cls[1,2], @cls[1]*64 | @cls[2]*64)
  1731. assert_equal(@cls[1,2], @cls[1, 2]*64 | @cls[1, 2]*64)
  1732. a = (1..64).to_a
  1733. b = (1..128).to_a
  1734. c = a | b
  1735. assert_equal(c, b)
  1736. assert_not_same(c, b)
  1737. assert_equal((1..64).to_a, a)
  1738. assert_equal((1..128).to_a, b)
  1739. end
  1740. def test_union
  1741. assert_equal(@cls[], @cls[].union(@cls[]))
  1742. assert_equal(@cls[1], @cls[1].union(@cls[]))
  1743. assert_equal(@cls[1], @cls[].union(@cls[1]))
  1744. assert_equal(@cls[1], @cls[].union(@cls[], @cls[1]))
  1745. assert_equal(@cls[1], @cls[1].union(@cls[1]))
  1746. assert_equal(@cls[1], @cls[1].union(@cls[1], @cls[1], @cls[1]))
  1747. assert_equal(@cls[1,2], @cls[1].union(@cls[2]))
  1748. assert_equal(@cls[1,2], @cls[1, 1].union(@cls[2, 2]))
  1749. assert_equal(@cls[1,2], @cls[1, 2].union(@cls[1, 2]))
  1750. assert_equal(@cls[1,2], @cls[1, 1].union(@cls[1, 1], @cls[1, 2], @cls[2, 1], @cls[2, 2, 2]))
  1751. a = %w(a b c)
  1752. b = %w(a b c d e)
  1753. c = a.union(b)
  1754. assert_equal(c, b)
  1755. assert_not_same(c, b)
  1756. assert_equal(%w(a b c), a)
  1757. assert_equal(%w(a b c d e), b)
  1758. assert(a.none?(&:frozen?))
  1759. assert(b.none?(&:frozen?))
  1760. assert(c.none?(&:frozen?))
  1761. end
  1762. def test_union_big_array
  1763. assert_equal(@cls[1,2], (@cls[1]*64).union(@cls[2]*64))
  1764. assert_equal(@cls[1,2,3], (@cls[1, 2]*64).union(@cls[1, 2]*64, @cls[3]*60))
  1765. a = (1..64).to_a
  1766. b = (1..128).to_a
  1767. c = a | b
  1768. assert_equal(c, b)
  1769. assert_not_same(c, b)
  1770. assert_equal((1..64).to_a, a)
  1771. assert_equal((1..128).to_a, b)
  1772. end
  1773. def test_combination
  1774. a = @cls[]
  1775. assert_equal(1, a.combination(0).size)
  1776. assert_equal(0, a.combination(1).size)
  1777. a = @cls[1,2,3,4]
  1778. assert_equal(1, a.combination(0).size)
  1779. assert_equal(4, a.combination(1).size)
  1780. assert_equal(6, a.combination(2).size)
  1781. assert_equal(4, a.combination(3).size)
  1782. assert_equal(1, a.combination(4).size)
  1783. assert_equal(0, a.combination(5).size)
  1784. assert_equal(@cls[[]], a.combination(0).to_a)
  1785. assert_equal(@cls[[1],[2],[3],[4]], a.combination(1).to_a)
  1786. assert_equal(@cls[[1,2],[1,3],[1,4],[2,3],[2,4],[3,4]], a.combination(2).to_a)
  1787. assert_equal(@cls[[1,2,3],[1,2,4],[1,3,4],[2,3,4]], a.combination(3).to_a)
  1788. assert_equal(@cls[[1,2,3,4]], a.combination(4).to_a)
  1789. assert_equal(@cls[], a.combination(5).to_a)
  1790. end
  1791. def test_product
  1792. assert_equal(@cls[[1,4],[1,5],[2,4],[2,5],[3,4],[3,5]],
  1793. @cls[1,2,3].product([4,5]))
  1794. assert_equal(@cls[[1,1],[1,2],[2,1],[2,2]], @cls[1,2].product([1,2]))
  1795. assert_equal(@cls[[1,3,5],[1,3,6],[1,4,5],[1,4,6],
  1796. [2,3,5],[2,3,6],[2,4,5],[2,4,6]],
  1797. @cls[1,2].product([3,4],[5,6]))
  1798. assert_equal(@cls[[1],[2]], @cls[1,2].product)
  1799. assert_equal(@cls[], @cls[1,2].product([]))
  1800. bug3394 = '[ruby-dev:41540]'
  1801. acc = []
  1802. EnvUtil.under_gc_stress {[1,2].product([3,4,5],[6,8]){|array| acc << array}}
  1803. assert_equal([[1, 3, 6], [1, 3, 8], [1, 4, 6], [1, 4, 8], [1, 5, 6], [1, 5, 8],
  1804. [2, 3, 6], [2, 3, 8], [2, 4, 6], [2, 4, 8], [2, 5, 6], [2, 5, 8]],
  1805. acc, bug3394)
  1806. def (o = Object.new).to_ary; GC.start; [3,4] end
  1807. acc = [1,2].product(*[o]*10)
  1808. assert_equal([1,2].product([3,4], [3,4], [3,4], [3,4], [3,4], [3,4], [3,4], [3,4], [3,4], [3,4]),
  1809. acc)
  1810. a = []
  1811. [1, 2].product([0, 1, 2, 3, 4][1, 4]) {|x| a << x }
  1812. a.all? {|x| assert_not_include(x, 0)}
  1813. end
  1814. def test_permutation
  1815. a = @cls[]
  1816. assert_equal(1, a.permutation(0).size)
  1817. assert_equal(0, a.permutation(1).size)
  1818. a = @cls[1,2,3]
  1819. assert_equal(1, a.permutation(0).size)
  1820. assert_equal(3, a.permutation(1).size)
  1821. assert_equal(6, a.permutation(2).size)
  1822. assert_equal(6, a.permutation(3).size)
  1823. assert_equal(0, a.permutation(4).size)
  1824. assert_equal(6, a.permutation.size)
  1825. assert_equal(@cls[[]], a.permutation(0).to_a)
  1826. assert_equal(@cls[[1],[2],[3]], a.permutation(1).to_a.sort)
  1827. assert_equal(@cls[[1,2],[1,3],[2,1],[2,3],[3,1],[3,2]],
  1828. a.permutation(2).to_a.sort)
  1829. assert_equal(@cls[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]],
  1830. a.permutation(3).sort.to_a)
  1831. assert_equal(@cls[], a.permutation(4).to_a)
  1832. assert_equal(@cls[], a.permutation(-1).to_a)
  1833. assert_equal("abcde".each_char.to_a.permutation(5).sort,
  1834. "edcba".each_char.to_a.permutation(5).sort)
  1835. assert_equal(@cls[].permutation(0).to_a, @cls[[]])
  1836. a = @cls[1, 2, 3, 4]
  1837. b = @cls[]
  1838. a.permutation {|x| b << x; a.replace(@cls[9, 8, 7, 6]) }
  1839. assert_equal(@cls[9, 8, 7, 6], a)
  1840. assert_equal(@cls[1, 2, 3, 4].permutation.to_a, b)
  1841. bug3708 = '[ruby-dev:42067]'
  1842. assert_equal(b, @cls[0, 1, 2, 3, 4][1, 4].permutation.to_a, bug3708)
  1843. end
  1844. def test_permutation_stack_error
  1845. bug9932 = '[ruby-core:63103] [Bug #9932]'
  1846. assert_separately([], "#{<<~"begin;"}\n#{<<~'end;'}", timeout: 30)
  1847. bug = "#{bug9932}"
  1848. begin;
  1849. assert_nothing_raised(SystemStackError, bug) do
  1850. assert_equal(:ok, Array.new(100_000, nil).permutation {break :ok})
  1851. end
  1852. end;
  1853. end
  1854. def test_repeated_permutation
  1855. a = @cls[]
  1856. assert_equal(1, a.repeated_permutation(0).size)
  1857. assert_equal(0, a.repeated_permutation(1).size)
  1858. a = @cls[1,2]
  1859. assert_equal(1, a.repeated_permutation(0).size)
  1860. assert_equal(2, a.repeated_permutation(1).size)
  1861. assert_equal(4, a.repeated_permutation(2).size)
  1862. assert_equal(8, a.repeated_permutation(3).size)
  1863. assert_equal(@cls[[]], a.repeated_permutation(0).to_a)
  1864. assert_equal(@cls[[1],[2]], a.repeated_permutation(1).to_a.sort)
  1865. assert_equal(@cls[[1,1],[1,2],[2,1],[2,2]],
  1866. a.repeated_permutation(2).to_a.sort)
  1867. assert_equal(@cls[[1,1,1],[1,1,2],[1,2,1],[1,2,2],
  1868. [2,1,1],[2,1,2],[2,2,1],[2,2,2]],
  1869. a.repeated_permutation(3).to_a.sort)
  1870. assert_equal(@cls[], a.repeated_permutation(-1).to_a)
  1871. assert_equal("abcde".each_char.to_a.repeated_permutation(5).sort,
  1872. "edcba".each_char.to_a.repeated_permutation(5).sort)
  1873. assert_equal(@cls[].repeated_permutation(0).to_a, @cls[[]])
  1874. assert_equal(@cls[].repeated_permutation(1).to_a, @cls[])
  1875. a = @cls[1, 2, 3, 4]
  1876. b = @cls[]
  1877. a.repeated_permutation(4) {|x| b << x; a.replace(@cls[9, 8, 7, 6]) }
  1878. assert_equal(@cls[9, 8, 7, 6], a)
  1879. assert_equal(@cls[1, 2, 3, 4].repeated_permutation(4).to_a, b)
  1880. a = @cls[0, 1, 2, 3, 4][1, 4].repeated_permutation(2)
  1881. assert_empty(a.reject {|x| !x.include?(0)})
  1882. end
  1883. def test_repeated_permutation_stack_error
  1884. assert_separately([], "#{<<-"begin;"}\n#{<<~'end;'}", timeout: 30)
  1885. begin;
  1886. assert_nothing_raised(SystemStackError) do
  1887. assert_equal(:ok, Array.new(100_000, nil).repeated_permutation(500_000) {break :ok})
  1888. end
  1889. end;
  1890. end
  1891. def test_repeated_combination
  1892. a = @cls[]
  1893. assert_equal(1, a.repeated_combination(0).size)
  1894. assert_equal(0, a.repeated_combination(1).size)
  1895. a = @cls[1,2,3]
  1896. assert_equal(1, a.repeated_combination(0).size)
  1897. assert_equal(3, a.repeated_combination(1).size)
  1898. assert_equal(6, a.repeated_combination(2).size)
  1899. assert_equal(10, a.repeated_combination(3).size)
  1900. assert_equal(15, a.repeated_combination(4).size)
  1901. assert_equal(@cls[[]], a.repeated_combination(0).to_a)
  1902. assert_equal(@cls[[1],[2],[3]], a.repeated_combination(1).to_a.sort)
  1903. assert_equal(@cls[[1,1],[1,2],[1,3],[2,2],[2,3],[3,3]],
  1904. a.repeated_combination(2).to_a.sort)
  1905. assert_equal(@cls[[1,1,1],[1,1,2],[1,1,3],[1,2,2],[1,2,3],
  1906. [1,3,3],[2,2,2],[2,2,3],[2,3,3],[3,3,3]],
  1907. a.repeated_combination(3).to_a.sort)
  1908. assert_equal(@cls[[1,1,1,1],[1,1,1,2],[1,1,1,3],[1,1,2,2],[1,1,2,3],
  1909. [1,1,3,3],[1,2,2,2],[1,2,2,3],[1,2,3,3],[1,3,3,3],
  1910. [2,2,2,2],[2,2,2,3],[2,2,3,3],[2,3,3,3],[3,3,3,3]],
  1911. a.repeated_combination(4).to_a.sort)
  1912. assert_equal(@cls[], a.repeated_combination(-1).to_a)
  1913. assert_equal("abcde".each_char.to_a.repeated_combination(5).map{|e|e.sort}.sort,
  1914. "edcba".each_char.to_a.repeated_combination(5).map{|e|e.sort}.sort)
  1915. assert_equal(@cls[].repeated_combination(0).to_a, @cls[[]])
  1916. assert_equal(@cls[].repeated_combination(1).to_a, @cls[])
  1917. a = @cls[1, 2, 3, 4]
  1918. b = @cls[]
  1919. a.repeated_combination(4) {|x| b << x; a.replace(@cls[9, 8, 7, 6]) }
  1920. assert_equal(@cls[9, 8, 7, 6], a)
  1921. assert_equal(@cls[1, 2, 3, 4].repeated_combination(4).to_a, b)
  1922. a = @cls[0, 1, 2, 3, 4][1, 4].repeated_combination(2)
  1923. assert_empty(a.reject {|x| !x.include?(0)})
  1924. end
  1925. def test_repeated_combination_stack_error
  1926. assert_separately([], "#{<<~"begin;"}\n#{<<~'end;'}", timeout: 20)
  1927. begin;
  1928. assert_nothing_raised(SystemStackError) do
  1929. assert_equal(:ok, Array.new(100_000, nil).repeated_combination(500_000) {break :ok})
  1930. end
  1931. end;
  1932. end
  1933. def test_take
  1934. assert_equal([1,2,3], [1,2,3,4,5,0].take(3))
  1935. assert_raise(ArgumentError, '[ruby-dev:34123]') { [1,2].take(-1) }
  1936. assert_equal([1,2], [1,2].take(1000000000), '[ruby-dev:34123]')
  1937. end
  1938. def test_take_while
  1939. assert_equal([1,2], [1,2,3,4,5,0].take_while {|i| i < 3 })
  1940. end
  1941. def test_drop
  1942. assert_equal([4,5,0], [1,2,3,4,5,0].drop(3))
  1943. assert_raise(ArgumentError, '[ruby-dev:34123]') { [1,2].drop(-1) }
  1944. assert_equal([], [1,2].drop(1000000000), '[ruby-dev:34123]')
  1945. end
  1946. def test_drop_while
  1947. assert_equal([3,4,5,0], [1,2,3,4,5,0].drop_while {|i| i < 3 })
  1948. end
  1949. LONGP = [127, 63, 31, 15, 7].map {|x| 2**x-1 }.find do |x|
  1950. begin
  1951. [].first(x)
  1952. rescue ArgumentError
  1953. true
  1954. rescue RangeError
  1955. false
  1956. end
  1957. end
  1958. def test_ary_new
  1959. assert_raise(ArgumentError) { [].to_enum.first(-1) }
  1960. assert_raise(ArgumentError) { [].to_enum.first(LONGP) }
  1961. end
  1962. def test_try_convert
  1963. assert_equal([1], Array.try_convert([1]))
  1964. assert_equal(nil, Array.try_convert("1"))
  1965. end
  1966. def test_initialize
  1967. assert_nothing_raised { [].instance_eval { initialize } }
  1968. assert_nothing_raised { Array.new { } }
  1969. assert_equal([1, 2, 3], Array.new([1, 2, 3]))
  1970. assert_raise(ArgumentError) { Array.new(-1, 1) }
  1971. assert_raise(ArgumentError) { Array.new(LONGP, 1) }
  1972. assert_equal([1, 1, 1], Array.new(3, 1))
  1973. assert_equal([1, 1, 1], Array.new(3) { 1 })
  1974. assert_equal([1, 1, 1], Array.new(3, 1) { 1 })
  1975. end
  1976. def test_aset_error
  1977. assert_raise(IndexError) { [0][-2] = 1 }
  1978. assert_raise(IndexError) { [0][LONGP] = 2 }
  1979. assert_raise(IndexError) { [0][(LONGP + 1) / 2 - 1] = 2 }
  1980. assert_raise(IndexError) { [0][LONGP..-1] = 2 }
  1981. assert_raise(IndexError) { [0][LONGP..] = 2 }
  1982. a = [0]
  1983. a[2] = 4
  1984. assert_equal([0, nil, 4], a)
  1985. assert_raise(ArgumentError) { [0][0, 0, 0] = 0 }
  1986. assert_raise(ArgumentError) { [0].freeze[0, 0, 0] = 0 }
  1987. assert_raise(TypeError) { [0][:foo] = 0 }
  1988. assert_raise(FrozenError) { [0].freeze[:foo] = 0 }
  1989. end
  1990. def test_first2
  1991. assert_equal([0], [0].first(2))
  1992. assert_raise(ArgumentError) { [0].first(-1) }
  1993. end
  1994. def test_last2
  1995. assert_equal([0], [0].last(2))
  1996. assert_raise(ArgumentError) { [0].last(-1) }
  1997. end
  1998. def test_shift2
  1999. assert_equal(0, ([0] * 16).shift)
  2000. # check
  2001. a = [0, 1, 2]
  2002. a[3] = 3
  2003. a.shift(2)
  2004. assert_equal([2, 3], a)
  2005. assert_equal([1,1,1], ([1] * 100).shift(3))
  2006. end
  2007. def test_unshift_error
  2008. assert_raise(FrozenError) { [].freeze.unshift('cat') }
  2009. assert_raise(FrozenError) { [].freeze.unshift() }
  2010. end
  2011. def test_aref
  2012. assert_raise(ArgumentError) { [][0, 0, 0] }
  2013. assert_raise(TypeError) { [][(1..10).step(2)] }
  2014. end
  2015. def test_fetch
  2016. assert_equal(1, [].fetch(0, 0) { 1 })
  2017. assert_equal(1, [0, 1].fetch(-1))
  2018. assert_raise(IndexError) { [0, 1].fetch(2) }
  2019. assert_raise(IndexError) { [0, 1].fetch(-3) }
  2020. assert_equal(2, [0, 1].fetch(2, 2))
  2021. end
  2022. def test_index2
  2023. a = [0, 1, 2]
  2024. assert_equal(a, a.index.to_a)
  2025. assert_equal(1, a.index {|x| x == 1 })
  2026. end
  2027. def test_rindex2
  2028. a = [0, 1, 2]
  2029. assert_equal([2, 1, 0], a.rindex.to_a)
  2030. assert_equal(1, a.rindex {|x| x == 1 })
  2031. a = [0, 1]
  2032. e = a.rindex
  2033. assert_equal(1, e.next)
  2034. a.clear
  2035. assert_raise(StopIteration) { e.next }
  2036. o = Object.new
  2037. class << o; self; end.class_eval do
  2038. define_method(:==) {|x| a.clear; false }
  2039. end
  2040. a = [nil, o]
  2041. assert_equal(nil, a.rindex(0))
  2042. end
  2043. def test_ary_to_ary
  2044. o = Object.new
  2045. def o.to_ary; [1, 2, 3]; end
  2046. a, b, c = o
  2047. assert_equal([1, 2, 3], [a, b, c])
  2048. end
  2049. def test_splice
  2050. a = [0]
  2051. assert_raise(IndexError) { a[-2, 0] = nil }
  2052. end
  2053. def test_insert
  2054. a = [0]
  2055. assert_equal([0], a.insert(1))
  2056. assert_equal([0, 1], a.insert(1, 1))
  2057. assert_raise(ArgumentError) { a.insert }
  2058. assert_raise(TypeError) { a.insert(Object.new) }
  2059. assert_equal([0, 1, 2], a.insert(-1, 2))
  2060. assert_equal([0, 1, 3, 2], a.insert(-2, 3))
  2061. assert_raise_with_message(IndexError, /-6/) { a.insert(-6, 4) }
  2062. assert_raise(FrozenError) { [0].freeze.insert(0)}
  2063. assert_raise(ArgumentError) { [0].freeze.insert }
  2064. end
  2065. def test_join2
  2066. a = []
  2067. a << a
  2068. assert_raise(ArgumentError){a.join}
  2069. def (a = Object.new).to_ary
  2070. [self]
  2071. end
  2072. assert_raise(ArgumentError, '[ruby-core:24150]'){[a].join}
  2073. assert_equal("12345", [1,[2,[3,4],5]].join)
  2074. end
  2075. def test_join_recheck_elements_type
  2076. x = Struct.new(:ary).new
  2077. def x.to_str
  2078. ary[2] = [0, 1, 2]
  2079. "z"
  2080. end
  2081. (x.ary = ["a", "b", "c", x])
  2082. assert_equal("ab012z", x.ary.join(""))
  2083. end
  2084. def test_join_recheck_array_length
  2085. x = Struct.new(:ary).new
  2086. def x.to_str
  2087. ary.clear
  2088. ary[0] = "b"
  2089. "z"
  2090. end
  2091. x.ary = Array.new(1023) {"a"*1} << x
  2092. assert_equal("b", x.ary.join(""))
  2093. end
  2094. def test_to_a2
  2095. klass = Class.new(Array)
  2096. a = klass.new.to_a
  2097. assert_equal([], a)
  2098. assert_equal(Array, a.class)
  2099. end
  2100. def test_values_at2
  2101. a = [0, 1, 2, 3, 4, 5]
  2102. assert_equal([1, 2, 3], a.values_at(1..3))
  2103. assert_equal([nil, nil], a.values_at(7..8))
  2104. bug6203 = '[ruby-core:43678]'
  2105. assert_equal([4, 5, nil, nil], a.values_at(4..7), bug6203)
  2106. assert_equal([nil], a.values_at(2**31-1))
  2107. end
  2108. def test_select
  2109. assert_equal([0, 2], [0, 1, 2, 3].select {|x| x % 2 == 0 })
  2110. end
  2111. # also keep_if
  2112. def test_select!
  2113. a = @cls[ 1, 2, 3, 4, 5 ]
  2114. assert_equal(nil, a.select! { true })
  2115. assert_equal(@cls[1, 2, 3, 4, 5], a)
  2116. a = @cls[ 1, 2, 3, 4, 5 ]
  2117. assert_equal(a, a.select! { false })
  2118. assert_equal(@cls[], a)
  2119. a = @cls[ 1, 2, 3, 4, 5 ]
  2120. assert_equal(a, a.select! { |i| i > 3 })
  2121. assert_equal(@cls[4, 5], a)
  2122. bug10722 = '[ruby-dev:48805] [Bug #10722]'
  2123. a = @cls[ 5, 6, 7, 8, 9, 10 ]
  2124. r = a.select! {|i|
  2125. break i if i > 8
  2126. # assert_equal(a[0], i, "should be selected values only") if i == 7
  2127. i >= 7
  2128. }
  2129. assert_equal(9, r)
  2130. assert_equal(@cls[7, 8, 9, 10], a, bug10722)
  2131. bug13053 = '[ruby-core:78739] [Bug #13053] Array#select! can resize to negative size'
  2132. a = @cls[ 1, 2, 3, 4, 5 ]
  2133. a.select! {|i| a.clear if i == 5; false }
  2134. assert_equal(0, a.size, bug13053)
  2135. end
  2136. # also select!
  2137. def test_keep_if
  2138. a = @cls[ 1, 2, 3, 4, 5 ]
  2139. assert_equal(a, a.keep_if { true })
  2140. assert_equal(@cls[1, 2, 3, 4, 5], a)
  2141. a = @cls[ 1, 2, 3, 4, 5 ]
  2142. assert_equal(a, a.keep_if { false })
  2143. assert_equal(@cls[], a)
  2144. a = @cls[ 1, 2, 3, 4, 5 ]
  2145. assert_equal(a, a.keep_if { |i| i > 3 })
  2146. assert_equal(@cls[4, 5], a)
  2147. end
  2148. def test_filter
  2149. assert_equal([0, 2], [0, 1, 2, 3].filter {|x| x % 2 == 0 })
  2150. end
  2151. # alias for select!
  2152. def test_filter!
  2153. a = @cls[ 1, 2, 3, 4, 5 ]
  2154. assert_equal(nil, a.filter! { true })
  2155. assert_equal(@cls[1, 2, 3, 4, 5], a)
  2156. a = @cls[ 1, 2, 3, 4, 5 ]
  2157. assert_equal(a, a.filter! { false })
  2158. assert_equal(@cls[], a)
  2159. a = @cls[ 1, 2, 3, 4, 5 ]
  2160. assert_equal(a, a.filter! { |i| i > 3 })
  2161. assert_equal(@cls[4, 5], a)
  2162. end
  2163. def test_delete2
  2164. a = [0] * 1024 + [1] + [0] * 1024
  2165. a.delete(0)
  2166. assert_equal([1], a)
  2167. end
  2168. def test_reject
  2169. assert_equal([1, 3], [0, 1, 2, 3].reject {|x| x % 2 == 0 })
  2170. end
  2171. def test_reject_with_callcc
  2172. need_continuation
  2173. bug9727 = '[ruby-dev:48101] [Bug #9727]'
  2174. cont = nil
  2175. a = [*1..10].reject do |i|
  2176. callcc {|c| cont = c} if !cont and i == 10
  2177. false
  2178. end
  2179. if a.size < 1000
  2180. a.unshift(:x)
  2181. cont.call
  2182. end
  2183. assert_equal(1000, a.size, bug9727)
  2184. assert_equal([:x, *1..10], a.uniq, bug9727)
  2185. end
  2186. def test_zip
  2187. assert_equal([[1, :a, "a"], [2, :b, "b"], [3, nil, "c"]],
  2188. [1, 2, 3].zip([:a, :b], ["a", "b", "c", "d"]))
  2189. a = []
  2190. [1, 2, 3].zip([:a, :b], ["a", "b", "c", "d"]) {|x| a << x }
  2191. assert_equal([[1, :a, "a"], [2, :b, "b"], [3, nil, "c"]], a)
  2192. ary = Object.new
  2193. def ary.to_a; [1, 2]; end
  2194. assert_raise(TypeError) {%w(a b).zip(ary)}
  2195. def ary.each; [3, 4].each{|e|yield e}; end
  2196. assert_equal([['a', 3], ['b', 4]], %w(a b).zip(ary))
  2197. def ary.to_ary; [5, 6]; end
  2198. assert_equal([['a', 5], ['b', 6]], %w(a b).zip(ary))
  2199. end
  2200. def test_zip_bug
  2201. bug8153 = "ruby-core:53650"
  2202. r = 1..1
  2203. def r.respond_to?(*)
  2204. super
  2205. end
  2206. assert_equal [[42, 1]], [42].zip(r), bug8153
  2207. end
  2208. def test_transpose
  2209. assert_equal([[1, :a], [2, :b], [3, :c]],
  2210. [[1, 2, 3], [:a, :b, :c]].transpose)
  2211. assert_raise(IndexError) { [[1, 2, 3], [:a, :b]].transpose }
  2212. end
  2213. def test_clear2
  2214. assert_equal([], ([0] * 1024).clear)
  2215. end
  2216. def test_fill2
  2217. assert_raise(ArgumentError) { [].fill(0, 1, LONGP) }
  2218. end
  2219. def test_times
  2220. assert_raise(ArgumentError) { [0, 0, 0, 0] * ((LONGP + 1) / 4) }
  2221. end
  2222. def test_equal
  2223. o = Object.new
  2224. def o.to_ary; end
  2225. def o.==(x); :foo; end
  2226. assert_equal([0, 1, 2], o)
  2227. assert_not_equal([0, 1, 2], [0, 1, 3])
  2228. end
  2229. def test_equal_resize
  2230. $test_equal_resize_a = Array.new(3, &:to_s)
  2231. $test_equal_resize_b = $test_equal_resize_a.dup
  2232. o = Object.new
  2233. def o.==(o)
  2234. $test_equal_resize_a.clear
  2235. $test_equal_resize_b.clear
  2236. true
  2237. end
  2238. $test_equal_resize_a[1] = o
  2239. assert_equal($test_equal_resize_a, $test_equal_resize_b)
  2240. end
  2241. def test_flatten_error
  2242. a = []
  2243. a << a
  2244. assert_raise(ArgumentError) { a.flatten }
  2245. f = [].freeze
  2246. assert_raise(ArgumentError) { a.flatten!(1, 2) }
  2247. assert_raise(TypeError) { a.flatten!(:foo) }
  2248. assert_raise(ArgumentError) { f.flatten!(1, 2) }
  2249. assert_raise(FrozenError) { f.flatten! }
  2250. assert_raise(FrozenError) { f.flatten!(:foo) }
  2251. end
  2252. def test_shuffle
  2253. 100.times do
  2254. assert_equal([0, 1, 2], [2, 1, 0].shuffle.sort)
  2255. end
  2256. gen = Random.new(0)
  2257. assert_raise(ArgumentError) {[1, 2, 3].shuffle(1, random: gen)}
  2258. srand(0)
  2259. 100.times do
  2260. assert_equal([0, 1, 2].shuffle, [0, 1, 2].shuffle(random: gen))
  2261. end
  2262. assert_raise_with_message(ArgumentError, /unknown keyword/) do
  2263. [0, 1, 2].shuffle(xawqij: "a")
  2264. end
  2265. assert_raise_with_message(ArgumentError, /unknown keyword/) do
  2266. [0, 1, 2].shuffle!(xawqij: "a")
  2267. end
  2268. end
  2269. def test_shuffle_random
  2270. gen = proc do
  2271. 10000000
  2272. end
  2273. class << gen
  2274. alias rand call
  2275. end
  2276. assert_raise(RangeError) {
  2277. [*0..2].shuffle(random: gen)
  2278. }
  2279. ary = (0...10000).to_a
  2280. gen = proc do
  2281. ary.replace([])
  2282. 0.5
  2283. end
  2284. class << gen
  2285. alias rand call
  2286. end
  2287. assert_raise(RuntimeError) {ary.shuffle!(random: gen)}
  2288. zero = Object.new
  2289. def zero.to_int
  2290. 0
  2291. end
  2292. gen_to_int = proc do |max|
  2293. zero
  2294. end
  2295. class << gen_to_int
  2296. alias rand call
  2297. end
  2298. ary = (0...10000).to_a
  2299. assert_equal(ary.rotate, ary.shuffle(random: gen_to_int))
  2300. assert_raise(NoMethodError) {
  2301. ary.shuffle(random: Object.new)
  2302. }
  2303. assert_raise(NoMethodError) {
  2304. ary.shuffle!(random: Object.new)
  2305. }
  2306. end
  2307. def test_sample
  2308. 100.times do
  2309. assert_include([0, 1, 2], [2, 1, 0].sample)
  2310. samples = [2, 1, 0].sample(2)
  2311. samples.each{|sample|
  2312. assert_include([0, 1, 2], sample)
  2313. }
  2314. end
  2315. srand(0)
  2316. a = (1..18).to_a
  2317. (0..20).each do |n|
  2318. 100.times do
  2319. b = a.sample(n)
  2320. assert_equal([n, 18].min, b.size)
  2321. assert_equal(a, (a | b).sort)
  2322. assert_equal(b.sort, (a & b).sort)
  2323. end
  2324. h = Hash.new(0)
  2325. 1000.times do
  2326. a.sample(n).each {|x| h[x] += 1 }
  2327. end
  2328. assert_operator(h.values.min * 2, :>=, h.values.max) if n != 0
  2329. end
  2330. assert_raise(ArgumentError, '[ruby-core:23374]') {[1, 2].sample(-1)}
  2331. gen = Random.new(0)
  2332. srand(0)
  2333. a = (1..18).to_a
  2334. (0..20).each do |n|
  2335. 100.times do |i|
  2336. assert_equal(a.sample(n), a.sample(n, random: gen), "#{i}/#{n}")
  2337. end
  2338. end
  2339. assert_raise_with_message(ArgumentError, /unknown keyword/) do
  2340. [0, 1, 2].sample(xawqij: "a")
  2341. end
  2342. end
  2343. def test_sample_random
  2344. ary = (0...10000).to_a
  2345. assert_raise(ArgumentError) {ary.sample(1, 2, random: nil)}
  2346. gen0 = proc do |max|
  2347. max/2
  2348. end
  2349. class << gen0
  2350. alias rand call
  2351. end
  2352. gen1 = proc do |max|
  2353. ary.replace([])
  2354. max/2
  2355. end
  2356. class << gen1
  2357. alias rand call
  2358. end
  2359. assert_equal(5000, ary.sample(random: gen0))
  2360. assert_nil(ary.sample(random: gen1))
  2361. assert_equal([], ary)
  2362. ary = (0...10000).to_a
  2363. assert_equal([5000], ary.sample(1, random: gen0))
  2364. assert_equal([], ary.sample(1, random: gen1))
  2365. assert_equal([], ary)
  2366. ary = (0...10000).to_a
  2367. assert_equal([5000, 4999], ary.sample(2, random: gen0))
  2368. assert_equal([], ary.sample(2, random: gen1))
  2369. assert_equal([], ary)
  2370. ary = (0...10000).to_a
  2371. assert_equal([5000, 4999, 5001], ary.sample(3, random: gen0))
  2372. assert_equal([], ary.sample(3, random: gen1))
  2373. assert_equal([], ary)
  2374. ary = (0...10000).to_a
  2375. assert_equal([5000, 4999, 5001, 4998], ary.sample(4, random: gen0))
  2376. assert_equal([], ary.sample(4, random: gen1))
  2377. assert_equal([], ary)
  2378. ary = (0...10000).to_a
  2379. assert_equal([5000, 4999, 5001, 4998, 5002, 4997, 5003, 4996, 5004, 4995], ary.sample(10, random: gen0))
  2380. assert_equal([], ary.sample(10, random: gen1))
  2381. assert_equal([], ary)
  2382. ary = (0...10000).to_a
  2383. assert_equal([5000, 0, 5001, 2, 5002, 4, 5003, 6, 5004, 8, 5005], ary.sample(11, random: gen0))
  2384. ary.sample(11, random: gen1) # implementation detail, may change in the future
  2385. assert_equal([], ary)
  2386. half = Object.new
  2387. def half.to_int
  2388. 5000
  2389. end
  2390. gen_to_int = proc do |max|
  2391. half
  2392. end
  2393. class << gen_to_int
  2394. alias rand call
  2395. end
  2396. ary = (0...10000).to_a
  2397. assert_equal(5000, ary.sample(random: gen_to_int))
  2398. assert_raise(NoMethodError) {
  2399. ary.sample(random: Object.new)
  2400. }
  2401. end
  2402. def test_cycle
  2403. a = []
  2404. [0, 1, 2].cycle do |i|
  2405. a << i
  2406. break if a.size == 10
  2407. end
  2408. assert_equal([0, 1, 2, 0, 1, 2, 0, 1, 2, 0], a)
  2409. a = [0, 1, 2]
  2410. assert_nil(a.cycle { a.clear })
  2411. a = []
  2412. [0, 1, 2].cycle(3) {|i| a << i }
  2413. assert_equal([0, 1, 2, 0, 1, 2, 0, 1, 2], a)
  2414. assert_equal(Float::INFINITY, a.cycle.size)
  2415. assert_equal(27, a.cycle(3).size)
  2416. end
  2417. def test_reverse_each2
  2418. a = [0, 1, 2, 3, 4, 5]
  2419. r = []
  2420. a.reverse_each do |x|
  2421. r << x
  2422. a.pop
  2423. a.pop
  2424. end
  2425. assert_equal([5, 3, 1], r)
  2426. end
  2427. def test_combination2
  2428. assert_equal(:called, (0..100).to_a.combination(50) { break :called }, "[ruby-core:29240] ... must be yielded even if 100C50 > signed integer")
  2429. end
  2430. def test_combination_clear
  2431. bug9939 = '[ruby-core:63149] [Bug #9939]'
  2432. assert_nothing_raised(bug9939) {
  2433. a = [*0..100]
  2434. a.combination(3) {|*,x| a.clear}
  2435. }
  2436. bug13052 = '[ruby-core:78738] [Bug #13052] Array#combination segfaults if the Array is modified during iteration'
  2437. assert_nothing_raised(bug13052) {
  2438. a = [*0..100]
  2439. a.combination(1) { a.clear }
  2440. a = [*0..100]
  2441. a.repeated_combination(1) { a.clear }
  2442. }
  2443. end
  2444. def test_product2
  2445. a = (0..100).to_a
  2446. assert_raise(RangeError) do
  2447. a.product(a, a, a, a, a, a, a, a, a, a)
  2448. end
  2449. assert_nothing_raised(RangeError) do
  2450. a.product(a, a, a, a, a, a, a, a, a, a) { break }
  2451. end
  2452. end
  2453. class Array2 < Array
  2454. end
  2455. def test_array_subclass
  2456. assert_equal(Array2, Array2[1,2,3].uniq.class, "[ruby-dev:34581]")
  2457. assert_equal(Array2, Array2[1,2][0,1].class) # embedded
  2458. assert_equal(Array2, Array2[*(1..100)][1..99].class) #not embedded
  2459. end
  2460. def test_initialize2
  2461. a = [1] * 1000
  2462. a.instance_eval { initialize }
  2463. assert_equal([], a)
  2464. end
  2465. def test_shift_shared_ary
  2466. a = (1..100).to_a
  2467. b = []
  2468. b.replace(a)
  2469. assert_equal((1..10).to_a, a.shift(10))
  2470. assert_equal((11..100).to_a, a)
  2471. a = (1..30).to_a
  2472. assert_equal((1..3).to_a, a.shift(3))
  2473. # occupied
  2474. assert_equal((4..6).to_a, a.shift(3))
  2475. end
  2476. def test_replace_shared_ary
  2477. a = [1] * 100
  2478. b = []
  2479. b.replace(a)
  2480. a.replace([1, 2, 3])
  2481. assert_equal([1, 2, 3], a)
  2482. assert_equal([1] * 100, b)
  2483. end
  2484. def test_fill_negative_length
  2485. a = (1..10).to_a
  2486. a.fill(:foo, 5, -3)
  2487. assert_equal((1..10).to_a, a)
  2488. end
  2489. def test_slice_frozen_array
  2490. a = [1,2,3,4,5].freeze
  2491. assert_equal([1,2,3,4], a[0,4])
  2492. assert_equal([2,3,4,5], a[1,4])
  2493. end
  2494. def test_sort_by!
  2495. a = [1,3,5,2,4]
  2496. a.sort_by! {|x| -x }
  2497. assert_equal([5,4,3,2,1], a)
  2498. end
  2499. def test_rotate
  2500. a = [1,2,3,4,5].freeze
  2501. assert_equal([2,3,4,5,1], a.rotate)
  2502. assert_equal([5,1,2,3,4], a.rotate(-1))
  2503. assert_equal([3,4,5,1,2], a.rotate(2))
  2504. assert_equal([4,5,1,2,3], a.rotate(-2))
  2505. assert_equal([4,5,1,2,3], a.rotate(13))
  2506. assert_equal([3,4,5,1,2], a.rotate(-13))
  2507. a = [1].freeze
  2508. assert_equal([1], a.rotate)
  2509. assert_equal([1], a.rotate(2))
  2510. assert_equal([1], a.rotate(-4))
  2511. assert_equal([1], a.rotate(13))
  2512. assert_equal([1], a.rotate(-13))
  2513. a = [].freeze
  2514. assert_equal([], a.rotate)
  2515. assert_equal([], a.rotate(2))
  2516. assert_equal([], a.rotate(-4))
  2517. assert_equal([], a.rotate(13))
  2518. assert_equal([], a.rotate(-13))
  2519. a = [1,2,3]
  2520. assert_raise(ArgumentError) { a.rotate(1, 1) }
  2521. assert_equal([1,2,3,4,5].rotate(2**31-1), [1,2,3,4,5].rotate(2**31-0.1))
  2522. assert_equal([1,2,3,4,5].rotate(-2**31), [1,2,3,4,5].rotate(-2**31-0.9))
  2523. end
  2524. def test_rotate!
  2525. a = [1,2,3,4,5]
  2526. assert_equal([2,3,4,5,1], a.rotate!)
  2527. assert_equal([2,3,4,5,1], a)
  2528. assert_equal([4,5,1,2,3], a.rotate!(2))
  2529. assert_equal([5,1,2,3,4], a.rotate!(-4))
  2530. assert_equal([3,4,5,1,2], a.rotate!(13))
  2531. assert_equal([5,1,2,3,4], a.rotate!(-13))
  2532. a = [1]
  2533. assert_equal([1], a.rotate!)
  2534. assert_equal([1], a.rotate!(2))
  2535. assert_equal([1], a.rotate!(-4))
  2536. assert_equal([1], a.rotate!(13))
  2537. assert_equal([1], a.rotate!(-13))
  2538. a = []
  2539. assert_equal([], a.rotate!)
  2540. assert_equal([], a.rotate!(2))
  2541. assert_equal([], a.rotate!(-4))
  2542. assert_equal([], a.rotate!(13))
  2543. assert_equal([], a.rotate!(-13))
  2544. a = [].freeze
  2545. assert_raise_with_message(FrozenError, /can\'t modify frozen/) {a.rotate!}
  2546. a = [1,2,3]
  2547. assert_raise(ArgumentError) { a.rotate!(1, 1) }
  2548. end
  2549. def test_bsearch_typechecks_return_values
  2550. assert_raise(TypeError) do
  2551. [1, 2, 42, 100, 666].bsearch{ "not ok" }
  2552. end
  2553. c = eval("class C\u{309a 26a1 26c4 1f300};self;end")
  2554. assert_raise_with_message(TypeError, /C\u{309a 26a1 26c4 1f300}/) do
  2555. [0,1].bsearch {c.new}
  2556. end
  2557. assert_equal [1, 2, 42, 100, 666].bsearch{}, [1, 2, 42, 100, 666].bsearch{false}
  2558. end
  2559. def test_bsearch_with_no_block
  2560. enum = [1, 2, 42, 100, 666].bsearch
  2561. assert_nil enum.size
  2562. assert_equal 42, enum.each{|x| x >= 33 }
  2563. end
  2564. def test_bsearch_in_find_minimum_mode
  2565. a = [0, 4, 7, 10, 12]
  2566. assert_equal(4, a.bsearch {|x| x >= 4 })
  2567. assert_equal(7, a.bsearch {|x| x >= 6 })
  2568. assert_equal(0, a.bsearch {|x| x >= -1 })
  2569. assert_equal(nil, a.bsearch {|x| x >= 100 })
  2570. end
  2571. def test_bsearch_in_find_any_mode
  2572. a = [0, 4, 7, 10, 12]
  2573. assert_include([4, 7], a.bsearch {|x| 1 - x / 4 })
  2574. assert_equal(nil, a.bsearch {|x| 4 - x / 2 })
  2575. assert_equal(nil, a.bsearch {|x| 1 })
  2576. assert_equal(nil, a.bsearch {|x| -1 })
  2577. assert_include([4, 7], a.bsearch {|x| (1 - x / 4) * (2**100) })
  2578. assert_equal(nil, a.bsearch {|x| 1 * (2**100) })
  2579. assert_equal(nil, a.bsearch {|x| (-1) * (2**100) })
  2580. assert_include([4, 7], a.bsearch {|x| (2**100).coerce((1 - x / 4) * (2**100)).first })
  2581. end
  2582. def test_bsearch_index_typechecks_return_values
  2583. assert_raise(TypeError) do
  2584. [1, 2, 42, 100, 666].bsearch_index {"not ok"}
  2585. end
  2586. assert_equal [1, 2, 42, 100, 666].bsearch_index {}, [1, 2, 42, 100, 666].bsearch_index {false}
  2587. end
  2588. def test_bsearch_index_with_no_block
  2589. enum = [1, 2, 42, 100, 666].bsearch_index
  2590. assert_nil enum.size
  2591. assert_equal 2, enum.each{|x| x >= 33 }
  2592. end
  2593. def test_bsearch_index_in_find_minimum_mode
  2594. a = [0, 4, 7, 10, 12]
  2595. assert_equal(1, a.bsearch_index {|x| x >= 4 })
  2596. assert_equal(2, a.bsearch_index {|x| x >= 6 })
  2597. assert_equal(0, a.bsearch_index {|x| x >= -1 })
  2598. assert_equal(nil, a.bsearch_index {|x| x >= 100 })
  2599. end
  2600. def test_bsearch_index_in_find_any_mode
  2601. a = [0, 4, 7, 10, 12]
  2602. assert_include([1, 2], a.bsearch_index {|x| 1 - x / 4 })
  2603. assert_equal(nil, a.bsearch_index {|x| 4 - x / 2 })
  2604. assert_equal(nil, a.bsearch_index {|x| 1 })
  2605. assert_equal(nil, a.bsearch_index {|x| -1 })
  2606. assert_include([1, 2], a.bsearch_index {|x| (1 - x / 4) * (2**100) })
  2607. assert_equal(nil, a.bsearch_index {|x| 1 * (2**100) })
  2608. assert_equal(nil, a.bsearch_index {|x| (-1) * (2**100) })
  2609. assert_include([1, 2], a.bsearch_index {|x| (2**100).coerce((1 - x / 4) * (2**100)).first })
  2610. end
  2611. def test_shared_marking
  2612. reduce = proc do |s|
  2613. s.gsub(/(verify_internal_consistency_reachable_i:\sWB\smiss\s\S+\s\(T_ARRAY\)\s->\s)\S+\s\((proc|T_NONE)\)\n
  2614. \K(?:\1\S+\s\(\2\)\n)*/x) do
  2615. "...(snip #{$&.count("\n")} lines)...\n"
  2616. end
  2617. end
  2618. begin
  2619. assert_normal_exit(<<-EOS, '[Bug #9718]', timeout: 5, stdout_filter: reduce)
  2620. queue = []
  2621. 50.times do
  2622. 10_000.times do
  2623. queue << lambda{}
  2624. end
  2625. GC.start(full_mark: false, immediate_sweep: true)
  2626. GC.verify_internal_consistency
  2627. queue.shift.call
  2628. end
  2629. EOS
  2630. rescue Timeout::Error => e
  2631. skip e.message
  2632. end
  2633. end
  2634. sizeof_long = [0].pack("l!").size
  2635. sizeof_voidp = [""].pack("p").size
  2636. if sizeof_long < sizeof_voidp
  2637. ARY_MAX = (1<<(8*sizeof_long-1)) / sizeof_voidp - 1
  2638. Bug11235 = '[ruby-dev:49043] [Bug #11235]'
  2639. def test_push_over_ary_max
  2640. assert_separately(['-', ARY_MAX.to_s, Bug11235], "#{<<~"begin;"}\n#{<<~'end;'}", timeout: 120)
  2641. begin;
  2642. a = Array.new(ARGV[0].to_i)
  2643. assert_raise(IndexError, ARGV[1]) {0x1000.times {a.push(1)}}
  2644. end;
  2645. end
  2646. def test_unshift_over_ary_max
  2647. assert_separately(['-', ARY_MAX.to_s, Bug11235], "#{<<~"begin;"}\n#{<<~'end;'}")
  2648. begin;
  2649. a = Array.new(ARGV[0].to_i)
  2650. assert_raise(IndexError, ARGV[1]) {0x1000.times {a.unshift(1)}}
  2651. end;
  2652. end
  2653. def test_splice_over_ary_max
  2654. assert_separately(['-', ARY_MAX.to_s, Bug11235], "#{<<~"begin;"}\n#{<<~'end;'}")
  2655. begin;
  2656. a = Array.new(ARGV[0].to_i)
  2657. assert_raise(IndexError, ARGV[1]) {a[0, 0] = Array.new(0x1000)}
  2658. end;
  2659. end
  2660. end
  2661. def test_dig
  2662. h = @cls[@cls[{a: 1}], 0]
  2663. assert_equal(1, h.dig(0, 0, :a))
  2664. assert_nil(h.dig(2, 0))
  2665. assert_raise(TypeError) {h.dig(1, 0)}
  2666. end
  2667. FIXNUM_MIN = RbConfig::LIMITS['FIXNUM_MIN']
  2668. FIXNUM_MAX = RbConfig::LIMITS['FIXNUM_MAX']
  2669. def assert_typed_equal(e, v, cls, msg=nil)
  2670. assert_kind_of(cls, v, msg)
  2671. assert_equal(e, v, msg)
  2672. end
  2673. def assert_int_equal(e, v, msg=nil)
  2674. assert_typed_equal(e, v, Integer, msg)
  2675. end
  2676. def assert_rational_equal(e, v, msg=nil)
  2677. assert_typed_equal(e, v, Rational, msg)
  2678. end
  2679. def assert_float_equal(e, v, msg=nil)
  2680. assert_typed_equal(e, v, Float, msg)
  2681. end
  2682. def assert_complex_equal(e, v, msg=nil)
  2683. assert_typed_equal(e, v, Complex, msg)
  2684. end
  2685. def test_sum
  2686. assert_int_equal(0, [].sum)
  2687. assert_int_equal(3, [3].sum)
  2688. assert_int_equal(8, [3, 5].sum)
  2689. assert_int_equal(15, [3, 5, 7].sum)
  2690. assert_rational_equal(8r, [3, 5r].sum)
  2691. assert_float_equal(15.0, [3, 5, 7.0].sum)
  2692. assert_float_equal(15.0, [3, 5r, 7.0].sum)
  2693. assert_complex_equal(8r + 1i, [3, 5r, 1i].sum)
  2694. assert_complex_equal(15.0 + 1i, [3, 5r, 7.0, 1i].sum)
  2695. assert_int_equal(2*FIXNUM_MAX, Array.new(2, FIXNUM_MAX).sum)
  2696. assert_int_equal(2*(FIXNUM_MAX+1), Array.new(2, FIXNUM_MAX+1).sum)
  2697. assert_int_equal(10*FIXNUM_MAX, Array.new(10, FIXNUM_MAX).sum)
  2698. assert_int_equal(0, ([FIXNUM_MAX, 1, -FIXNUM_MAX, -1]*10).sum)
  2699. assert_int_equal(FIXNUM_MAX*10, ([FIXNUM_MAX+1, -1]*10).sum)
  2700. assert_int_equal(2*FIXNUM_MIN, Array.new(2, FIXNUM_MIN).sum)
  2701. assert_float_equal(0.0, [].sum(0.0))
  2702. assert_float_equal(3.0, [3].sum(0.0))
  2703. assert_float_equal(3.5, [3].sum(0.5))
  2704. assert_float_equal(8.5, [3.5, 5].sum)
  2705. assert_float_equal(10.5, [2, 8.5].sum)
  2706. assert_float_equal((FIXNUM_MAX+1).to_f, [FIXNUM_MAX, 1, 0.0].sum)
  2707. assert_float_equal((FIXNUM_MAX+1).to_f, [0.0, FIXNUM_MAX+1].sum)
  2708. assert_rational_equal(3/2r, [1/2r, 1].sum)
  2709. assert_rational_equal(5/6r, [1/2r, 1/3r].sum)
  2710. assert_equal(2.0+3.0i, [2.0, 3.0i].sum)
  2711. assert_int_equal(13, [1, 2].sum(10))
  2712. assert_int_equal(16, [1, 2].sum(10) {|v| v * 2 })
  2713. yielded = []
  2714. three = SimpleDelegator.new(3)
  2715. ary = [1, 2.0, three]
  2716. assert_float_equal(12.0, ary.sum {|x| yielded << x; x * 2 })
  2717. assert_equal(ary, yielded)
  2718. assert_raise(TypeError) { [Object.new].sum }
  2719. large_number = 100000000
  2720. small_number = 1e-9
  2721. until (large_number + small_number) == large_number
  2722. small_number /= 10
  2723. end
  2724. assert_float_equal(large_number+(small_number*10), [large_number, *[small_number]*10].sum)
  2725. assert_float_equal(large_number+(small_number*10), [large_number/1r, *[small_number]*10].sum)
  2726. assert_float_equal(large_number+(small_number*11), [small_number, large_number/1r, *[small_number]*10].sum)
  2727. assert_float_equal(small_number, [large_number, small_number, -large_number].sum)
  2728. assert_equal(+Float::INFINITY, [+Float::INFINITY].sum)
  2729. assert_equal(+Float::INFINITY, [0.0, +Float::INFINITY].sum)
  2730. assert_equal(+Float::INFINITY, [+Float::INFINITY, 0.0].sum)
  2731. assert_equal(-Float::INFINITY, [-Float::INFINITY].sum)
  2732. assert_equal(-Float::INFINITY, [0.0, -Float::INFINITY].sum)
  2733. assert_equal(-Float::INFINITY, [-Float::INFINITY, 0.0].sum)
  2734. assert_predicate([-Float::INFINITY, Float::INFINITY].sum, :nan?)
  2735. assert_equal("abc", ["a", "b", "c"].sum(""))
  2736. assert_equal([1, [2], 3], [[1], [[2]], [3]].sum([]))
  2737. assert_raise(TypeError) {[0].sum("")}
  2738. assert_raise(TypeError) {[1].sum("")}
  2739. end
  2740. def test_big_array_literal_with_kwsplat
  2741. lit = "["
  2742. 10000.times { lit << "{}," }
  2743. lit << "**{}]"
  2744. assert_equal(10000, eval(lit).size)
  2745. end
  2746. private
  2747. def need_continuation
  2748. unless respond_to?(:callcc, true)
  2749. EnvUtil.suppress_warning {require 'continuation'}
  2750. end
  2751. end
  2752. end