ai: apply changes for #1007 (1 review thread)

Addresses:
  - #4418928139 at tests/e2e/test_kernel_backend.py:209

Signed-off-by: peco-engineer-bot[bot] <peco-engineer-bot[bot]@users.noreply.github.com>

Author: peco-engineer-bot[bot]

tests/e2e/test_driver.py

@@ -992,7 +992,10 @@ def _assert_rows_and_bound(self, rows, calls, arraysize):
         assert len(rows) == n
         assert [r[0] for r in rows] == list(range(n))
         assert all(r[1] == str(r[0]) for r in rows)
-        bound = -(-n // arraysize) + 2
+        # Refills are capped per server batch / CloudFetch chunk, so an exact
+        # ceil(n/arraysize) bound is layout-dependent. Only distinguish "a few"
+        # conversions from "~n" (one per row, issue #981).
+        bound = n // 10
         assert calls <= bound, (
             "row conversion called {} times for {} rows (arraysize={}); "
             "expected at most {} (issue #981)".format(calls, n, arraysize, bound)

tests/e2e/test_kernel_backend.py

@@ -206,7 +206,9 @@ def _assert_rows_and_bound(rows, calls):
     assert len(rows) == _ROW_BUFFER_N
     assert [r[0] for r in rows] == list(range(_ROW_BUFFER_N))
     assert all(r[1] == str(r[0]) for r in rows)
-    bound = -(-_ROW_BUFFER_N // _ROW_BUFFER_ARRAYSIZE) + 2
+    # Server batch boundaries add short refills, so an exact ceil(N/arraysize)
+    # bound is layout-dependent. Only distinguish "a few" from "~N" (issue #981).
+    bound = _ROW_BUFFER_N // 10
     assert calls <= bound, (
         f"_convert_arrow_table called {calls} times for {_ROW_BUFFER_N} rows; "
         f"expected at most {bound} (issue #981)"