Test config selection modes

Author: OptHuang

README.md

@@ -35,6 +35,7 @@ The `CI` workflow runs daily and on pushes. It checks the OptiProfiler adapter l
 - selecting a small set of representative `u`, `b`, `l`, and `n` problems;
 - loading each selected problem through `s2mpj_load`;
 - evaluating `fun`, `cub`, and `ceq` at the initial point;
+- checking `variable_size` and `test_feasibility_problems` environment overrides;
 - sampling a few additional small problems each day with at most two numerical-library threads.
 
 Locally, from this repository:

tests/test_s2mpj_python.py

@@ -2,6 +2,7 @@
 
 from datetime import date
 from pathlib import Path
+from contextlib import contextmanager
 import math
 import os
 import random
@@ -30,6 +31,24 @@
 REPRESENTATIVE_PROBLEMS = ["ALLINITU", "ALLINIT", "ALSOTAME", "ALLINITA"]
 
 
+@contextmanager
+def _temporary_env(**updates):
+    previous = {key: os.environ.get(key) for key in updates}
+    try:
+        for key, value in updates.items():
+            if value is None:
+                os.environ.pop(key, None)
+            else:
+                os.environ[key] = str(value)
+        yield
+    finally:
+        for key, value in previous.items():
+            if value is None:
+                os.environ.pop(key, None)
+            else:
+                os.environ[key] = value
+
+
 def _as_array(value):
     if value is None:
         return np.empty(0)
@@ -96,6 +115,64 @@ def test_daily_random_small_problem_sample(self):
             with self.subTest(problem=problem_name):
                 _assert_problem_contract(self, problem_name)
 
+    def test_config_environment_overrides_variable_size(self):
+        options = {
+            "ptype": "ubln",
+            "maxdim": 5,
+            "maxb": 20,
+            "maxlcon": 20,
+            "maxnlcon": 20,
+            "maxcon": 20,
+        }
+        with _temporary_env(
+            S2MPJ_VARIABLE_SIZE=None,
+            S2MPJ_TEST_FEASIBILITY_PROBLEMS=None,
+        ):
+            default_names = s2mpj_select(dict(options))
+        with _temporary_env(
+            S2MPJ_VARIABLE_SIZE="all",
+            S2MPJ_TEST_FEASIBILITY_PROBLEMS="0",
+        ):
+            all_names = s2mpj_select(dict(options))
+
+        self.assertGreater(len(all_names), len(default_names))
+        self.assertIn("CHEBYQAD_2", all_names)
+        self.assertNotIn("CHEBYQAD_2", default_names)
+
+    def test_config_environment_overrides_feasibility_selection(self):
+        options = {
+            "ptype": "ubln",
+            "maxdim": 5,
+            "maxb": 20,
+            "maxlcon": 20,
+            "maxnlcon": 20,
+            "maxcon": 20,
+        }
+        with _temporary_env(
+            S2MPJ_VARIABLE_SIZE="default",
+            S2MPJ_TEST_FEASIBILITY_PROBLEMS="1",
+        ):
+            feasibility_names = s2mpj_select(dict(options))
+        with _temporary_env(
+            S2MPJ_VARIABLE_SIZE="default",
+            S2MPJ_TEST_FEASIBILITY_PROBLEMS="2",
+        ):
+            all_names = s2mpj_select(dict(options))
+
+        self.assertIn("ARGAUSS", feasibility_names)
+        self.assertNotIn("ALLINITU", feasibility_names)
+        self.assertIn("ARGAUSS", all_names)
+        self.assertIn("ALLINITU", all_names)
+
+    def test_config_environment_rejects_invalid_values(self):
+        options = {"ptype": "u", "maxdim": 5}
+        with _temporary_env(S2MPJ_VARIABLE_SIZE="not-a-mode"):
+            with self.assertRaises(ValueError):
+                s2mpj_select(dict(options))
+        with _temporary_env(S2MPJ_TEST_FEASIBILITY_PROBLEMS="3"):
+            with self.assertRaises(ValueError):
+                s2mpj_select(dict(options))
+
 
 if __name__ == "__main__":
     unittest.main()