Function test_update_tracker

tests/test_logger.py:218–246

Test the _update_tracker method.

()

def test_update_tracker():
    """Test the _update_tracker method."""
    optimizer = BayesianOptimization(target_func, PBOUNDS, random_state=1)
    logger = ScreenLogger()

    # Initial state
    assert logger._iterations == 0
    assert logger._previous_max is None

    # Update with first observation
    optimizer.register(params={"p1": 1, "p2": 2}, target=3)
    logger._update_tracker(optimizer.max)
    assert logger._iterations == 1
    assert logger._previous_max == 3
    assert logger._previous_max_params == {"p1": 1, "p2": 2}

    # Update with lower observation
    optimizer.register(params={"p1": 0.5, "p2": 1}, target=1.5)
    logger._update_tracker(optimizer.max)
    assert logger._iterations == 2
    assert logger._previous_max == 3  # Unchanged
    assert logger._previous_max_params == {"p1": 1, "p2": 2}  # Unchanged

    # Update with higher observation
    optimizer.register(params={"p1": 2, "p2": 2}, target=4)
    logger._update_tracker(optimizer.max)
    assert logger._iterations == 3
    assert logger._previous_max == 4  # Updated
    assert logger._previous_max_params == {"p1": 2, "p2": 2}  # Updated


@patch("sys.stdout", new_callable=io.StringIO)