From a8e7fee80129b0ba360c2671582117c8e18a6464 Mon Sep 17 00:00:00 2001
From: Takashi Kokubun <takashikkbn@gmail.com>
Date: Thu, 16 Mar 2023 09:22:42 -0700
Subject: [PATCH] Revert "core_assertions.rb: Refine
 `assert_linear_performance`"

This reverts commit cae4342dd559e34c1ce6219593f77f0ad80286da.

This is failing a lot of CIs and nobody is actively looking into fixing
it. Let me revert this until we have a solution to it.
---
 test/ruby/test_regexp.rb    |  2 +-
 tool/lib/core_assertions.rb | 27 +++++++++++----------------
 2 files changed, 12 insertions(+), 17 deletions(-)

diff --git a/test/ruby/test_regexp.rb b/test/ruby/test_regexp.rb
index 81dac20648761e..a2997af1060b45 100644
--- a/test/ruby/test_regexp.rb
+++ b/test/ruby/test_regexp.rb
@@ -1783,7 +1783,7 @@ def test_linear_time_p
 
   def test_linear_performance
     pre = ->(n) {[Regexp.new("a?" * n + "a" * n), "a" * n]}
-    assert_linear_performance([10, 29], pre: pre) do |re, s|
+    assert_linear_performance(factor: 29, first: 10, max: 1, pre: pre) do |re, s|
       re =~ s
     end
   end
diff --git a/tool/lib/core_assertions.rb b/tool/lib/core_assertions.rb
index c195c91482dfc5..119deccb9ff6ba 100644
--- a/tool/lib/core_assertions.rb
+++ b/tool/lib/core_assertions.rb
@@ -738,28 +738,23 @@ def assert_all_assertions_foreach(msg = nil, *keys, &block)
       end
       alias all_assertions_foreach assert_all_assertions_foreach
 
-      # 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
+      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
           st = Process.clock_gettime(Process::CLOCK_MONOTONIC)
-          yield(*arg)
+          yield arg
           (Process.clock_gettime(Process::CLOCK_MONOTONIC) - st)
         end.max
 
-        seq.each do |i|
-          next if i == first
-          t = tmax * i / first
-          *arg = pre.call(i)
-          message = "[#{i}]: in #{t}s"
+        (first >= factor ? 2 : 1).upto(max) do |i|
+          n = i * factor
+          t = tmax * factor
+          arg = pre.call(n)
+          message = "[#{i}]: #{n} 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