add tests to check params_config implementation

Author: adrianmolzon

tests/test_logger.py

@@ -1,8 +1,9 @@
 from __future__ import annotations
 
 import io
-from unittest.mock import patch
+from unittest.mock import MagicMock, patch
 
+import pytest
 from colorama import Fore
 
 from bayes_opt import BayesianOptimization
@@ -51,6 +52,44 @@ def test_is_constrained_property():
     assert logger.is_constrained
 
 
+def test_params_config_property():
+    """Test the params_config property getter and setter."""
+    # Test the getter with default initialization (None)
+    logger = ScreenLogger()
+    assert logger.params_config is None
+
+    # Test initialization with a params_config
+    mock_config = {"param1": MagicMock(), "param2": MagicMock()}
+    logger_with_config = ScreenLogger(params_config=mock_config)
+    assert logger_with_config.params_config is mock_config
+
+    # Test the setter
+    new_config = {"param3": MagicMock(), "param4": MagicMock()}
+    logger.params_config = new_config
+    assert logger.params_config is new_config
+
+    # Test that the logger actually uses the params_config
+    optimizer = BayesianOptimization(target_func, PBOUNDS, random_state=1)
+    logger.params_config = optimizer._space._params_config
+    optimizer.register(params={"p1": 1.5, "p2": 2.5}, target=4.0)
+
+    # This should not raise an error now that params_config is set
+    step_str = logger._print_step(optimizer._space.res()[-1], optimizer._space.keys)
+    assert "|" in step_str
+    assert "1" in step_str  # iteration
+    assert "4.0" in step_str  # target value
+
+
+def test_print_step_without_params_config():
+    """Test that _print_step raises an error when params_config is None."""
+    logger = ScreenLogger()
+    optimizer = BayesianOptimization(target_func, PBOUNDS, random_state=1)
+    optimizer.register(params={"p1": 1.5, "p2": 2.5}, target=4.0)
+
+    with pytest.raises(ValueError, match="Parameter configuration is not set"):
+        logger._print_step(optimizer._space.res()[-1], optimizer._space.keys)
+
+
 def test_format_number():
     """Test the _format_number method."""
     logger = ScreenLogger()