TST: Speed-up SMBO tests

Author: kernc

sambo/_smbo.py

@@ -339,7 +339,8 @@ def ask(
         assert isinstance(kappa, (Real, Iterable)), kappa
         self._init_once()
 
-        n_points = min(80_000, 20_000 * int(len(self.bounds)**2))  # TODO: Make this a param?
+        n_points = min(self.MAX_POINTS_PER_ITER,
+                       self.POINTS_PER_DIM * int(len(self.bounds)**2))  # TODO: Make this a param?
         nfev = len(self._X)
         if nfev < 10 * len(self.bounds)**2:
             X = _sample_population(self.bounds, n_points, self.constraints, self.rng)
@@ -363,6 +364,9 @@ def ask(
         self._X_ask.extend(map(tuple, X))
         return X
 
+    POINTS_PER_DIM = 20_000
+    MAX_POINTS_PER_ITER = 80_000
+
     def tell(self, y: float | list[float],
              x: Optional[float | tuple[float] | list[tuple[float]]] = None):
         """

sambo/_test.py

@@ -6,6 +6,7 @@
 from pathlib import Path
 from pprint import pprint
 from typing import get_type_hints
+from unittest.mock import patch
 
 import numpy as np
 from matplotlib import pyplot as plt
@@ -36,6 +37,9 @@
 BUILTIN_METHODS = ['shgo', 'sceua', 'smbo']
 BUILTIN_ESTIMATORS = ['gp', 'et', 'gb']
 
+Optimizer.POINTS_PER_DIM = 1_000
+Optimizer.MAX_POINTS_PER_ITER = 10_000
+
 
 def check_result(res, y_true, atol=1e-5):
     np.testing.assert_allclose(res.fun, y_true, atol=atol, err_msg=res)
@@ -214,7 +218,7 @@ def test_sceua(self):
 
     def test_smbo(self):
         res = minimize(**ROSEN_TEST_PARAMS, method='smbo', max_iter=20, estimator='gp')
-        check_result(res, 0, atol=5)
+        check_result(res, 0, atol=11)
 
     def test_args(self):
         def f(x, a):
@@ -399,6 +403,8 @@ def test_website_example1(self):
         print(type(res), res, sep='\n\n')
         self.assertAlmostEqual(res.fun, 0, places=0, msg=res)
 
+    @patch.object(Optimizer, 'POINTS_PER_DIM', 20_000)
+    @patch.object(Optimizer, 'MAX_POINTS_PER_ITER', 80_000)
     def test_website_example2(self):
 
         def evaluate(x):