core_assertions.rb: Refine `assert_linear_performance`

* Use an `Enumerable` as factors, instead of three arguments.
This commit is contained in:
Nobuyoshi Nakada 2023-03-16 17:19:01 +09:00
Родитель 5cffa69c1b
Коммит cae4342dd5
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 7CD2805BFA3770C6
2 изменённых файлов: 17 добавлений и 12 удалений

Просмотреть файл

@ -1783,7 +1783,7 @@ class TestRegexp < Test::Unit::TestCase
def test_linear_performance
pre = ->(n) {[Regexp.new("a?" * n + "a" * n), "a" * n]}
assert_linear_performance(factor: 29, first: 10, max: 1, pre: pre) do |re, s|
assert_linear_performance([10, 29], pre: pre) do |re, s|
re =~ s
end
end

Просмотреть файл

@ -738,23 +738,28 @@ eom
end
alias all_assertions_foreach assert_all_assertions_foreach
def assert_linear_performance(factor: 10_000, first: factor, max: 2, rehearsal: first, pre: ->(n) {n})
n = first
arg = pre.call(n)
tmax = (0..rehearsal).map do
# Expect +seq+ to respond to +first+ and +each+ methods, e.g.,
# Array, Range, Enumerator::ArithmeticSequence and other
# Enumerable-s, and each elements should be size factors.
#
# :yield: each elements of +seq+.
def assert_linear_performance(seq, rehearsal: nil, pre: ->(n) {n})
first = seq.first
*arg = pre.call(first)
tmax = (0..(rehearsal || first)).map do
st = Process.clock_gettime(Process::CLOCK_MONOTONIC)
yield arg
yield(*arg)
(Process.clock_gettime(Process::CLOCK_MONOTONIC) - st)
end.max
(first >= factor ? 2 : 1).upto(max) do |i|
n = i * factor
t = tmax * factor
arg = pre.call(n)
message = "[#{i}]: #{n} in #{t}s"
seq.each do |i|
next if i == first
t = tmax * i / first
*arg = pre.call(i)
message = "[#{i}]: in #{t}s"
Timeout.timeout(t, nil, message) do
st = Process.clock_gettime(Process::CLOCK_MONOTONIC)
yield arg
yield(*arg)
assert_operator (Process.clock_gettime(Process::CLOCK_MONOTONIC) - st), :<=, t, message
end
end