From 6d670d7ea85d7bbbf8e3a0a31d59b6b1396b580e Mon Sep 17 00:00:00 2001
From: anakin87 <stefanofiorucci@gmail.com>
Date: Thu, 23 Apr 2026 15:50:00 +0200
Subject: [PATCH] test: optimum - add unit tests for configs

---
 integrations/optimum/tests/test_configs.py | 51 ++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 integrations/optimum/tests/test_configs.py

diff --git a/integrations/optimum/tests/test_configs.py b/integrations/optimum/tests/test_configs.py
new file mode 100644
index 0000000000..70508a3a79
--- /dev/null
+++ b/integrations/optimum/tests/test_configs.py
@@ -0,0 +1,51 @@
+# SPDX-FileCopyrightText: 2024-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import pytest
+from optimum.onnxruntime.configuration import OptimizationConfig, QuantizationConfig
+
+from haystack_integrations.components.embedders.optimum.optimization import (
+    OptimumEmbedderOptimizationConfig,
+    OptimumEmbedderOptimizationMode,
+)
+from haystack_integrations.components.embedders.optimum.quantization import (
+    OptimumEmbedderQuantizationConfig,
+    OptimumEmbedderQuantizationMode,
+)
+
+
+class TestOptimumEmbedderQuantizationConfig:
+    @pytest.mark.parametrize(
+        "mode",
+        [
+            OptimumEmbedderQuantizationMode.ARM64,
+            OptimumEmbedderQuantizationMode.AVX2,
+            OptimumEmbedderQuantizationMode.AVX512,
+            OptimumEmbedderQuantizationMode.AVX512_VNNI,
+        ],
+    )
+    def test_to_optimum_config_returns_quantization_config_for_each_mode(self, mode):
+        config = OptimumEmbedderQuantizationConfig(mode=mode, per_channel=True)
+        optimum_config = config.to_optimum_config()
+
+        assert isinstance(optimum_config, QuantizationConfig)
+        assert optimum_config.is_static is False
+        assert optimum_config.per_channel is True
+
+
+class TestOptimumEmbedderOptimizationConfig:
+    @pytest.mark.parametrize(
+        "mode",
+        [
+            OptimumEmbedderOptimizationMode.O1,
+            OptimumEmbedderOptimizationMode.O2,
+            OptimumEmbedderOptimizationMode.O3,
+            OptimumEmbedderOptimizationMode.O4,
+        ],
+    )
+    def test_to_optimum_config_returns_optimization_config_for_each_mode(self, mode):
+        config = OptimumEmbedderOptimizationConfig(mode=mode, for_gpu=False)
+        optimum_config = config.to_optimum_config()
+
+        assert isinstance(optimum_config, OptimizationConfig)