Add tests for new composite kernels

JamesBagley · May 16, 2024 · 7f3d57c · 7f3d57c
1 parent 1f432e1
commit 7f3d57c
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/tests/test_iterations.py b/tests/test_iterations.py
@@ -1,12 +1,11 @@
 # TODO: This file needs to be refactored.
 """Tests various configurations for a small number of iterations."""
 
-
 import pytest
 
 from baybe.acquisition.base import AcquisitionFunction
 from baybe.kernels.basic import MaternKernel
-from baybe.kernels.composite import ScaleKernel
+from baybe.kernels.composite import AdditiveKernel, ProductKernel, ScaleKernel
 from baybe.priors import (
     GammaPrior,
     HalfCauchyPrior,
@@ -140,7 +139,12 @@
     for prior in valid_priors
 ]
 
-valid_kernels = valid_base_kernels + valid_scale_kernels
+valid_composite_kernels = [
+    AdditiveKernel([MaternKernel(1.5), MaternKernel(2.5)]),
+    ProductKernel([MaternKernel(1.5), MaternKernel(2.5)]),
+]
+
+valid_kernels = valid_base_kernels + valid_scale_kernels + valid_composite_kernels
 
 test_targets = [
     ["Target_max"],