feat: add coverage for add_extra_session_options (#187)

* test: add coverage for add_extra_session_options

Co-authored-by: n24q02m <135627235+n24q02m@users.noreply.github.com>

* style: format tests with ruff

Co-authored-by: n24q02m <135627235+n24q02m@users.noreply.github.com>

---------

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>

Author: n24q02m

tests/test_onnx_model_loading.py

@@ -218,3 +218,22 @@ def test_load_cuda_warning(model: ConcreteOnnxModel, mock_ort):
 
     with pytest.warns(RuntimeWarning, match="Attempt to set CUDAExecutionProvider failed"):
         model._load_onnx_model(Path("dummy"), "model.onnx", threads=None, cuda=True)
+
+
+def test_add_extra_session_options():
+    session_options = MagicMock()
+
+    # Test valid options
+    ConcreteOnnxModel.add_extra_session_options(session_options, {"enable_cpu_mem_arena": False})
+    assert session_options.enable_cpu_mem_arena is False
+
+    # Test valid options (True)
+    ConcreteOnnxModel.add_extra_session_options(session_options, {"enable_cpu_mem_arena": True})
+    assert session_options.enable_cpu_mem_arena is True
+
+    # Test invalid option
+    with pytest.raises(
+        ValueError,
+        match="invalid_option is unknown or not exposed \\(exposed options: \\('enable_cpu_mem_arena',\\)\\)",
+    ):
+        ConcreteOnnxModel.add_extra_session_options(session_options, {"invalid_option": True})