| Total Complexity | 2 |
| Total Lines | 43 |
| Duplicated Lines | 0 % |
| Changes | 0 | ||
| 1 | """Test module for optimizer memory functionality.""" |
||
| 2 | |||
| 3 | import numpy as np |
||
| 4 | import pytest |
||
| 5 | |||
| 6 | from hyperactive import Hyperactive |
||
| 7 | |||
| 8 | from ._parametrize import optimizers |
||
| 9 | |||
| 10 | |||
| 11 | def objective_function(opt): |
||
| 12 | """Return simple quadratic objective function for memory testing.""" |
||
| 13 | score = -opt["x1"] * opt["x1"] |
||
| 14 | return score |
||
| 15 | |||
| 16 | |||
| 17 | search_space = {"x1": list(np.arange(-10, 11, 1))} |
||
| 18 | |||
| 19 | |||
| 20 | @pytest.mark.parametrize(*optimizers) |
||
| 21 | def test_memory_0(Optimizer): |
||
| 22 | """Test memory functionality with multiple search runs.""" |
||
| 23 | optimizer = Optimizer() |
||
| 24 | |||
| 25 | n_iter = 30 |
||
| 26 | |||
| 27 | hyper = Hyperactive() |
||
| 28 | hyper.add_search( |
||
| 29 | objective_function, |
||
| 30 | search_space, |
||
| 31 | optimizer=optimizer, |
||
| 32 | n_iter=n_iter, |
||
| 33 | n_jobs=2, |
||
| 34 | ) |
||
| 35 | hyper.add_search( |
||
| 36 | objective_function, |
||
| 37 | search_space, |
||
| 38 | optimizer=optimizer, |
||
| 39 | n_iter=n_iter, |
||
| 40 | n_jobs=2, |
||
| 41 | ) |
||
| 42 | hyper.run() |
||
| 43 |