test cli dispatch

2026-08-29 22:20:36 +02:00 · 2026-08-19 22:25:38 -04:00
parent 3dbe90d9a9
commit 94d65f8769
1 changed files with 16 additions and 0 deletions
@@ -162,3 +162,19 @@ class TestMainCLI:
            sys.argv = old_argv

        mock_check.assert_called_once()
+
+    @patch("obliteratus.capability_check._run_lm_eval")
+    def test_cli_dispatch(self, mock_lm_eval, tmp_path):
+        """Test that 'obliteratus capability-check' dispatches correctly."""
+        from obliteratus.cli import main as cli_main
+
+        mock_lm_eval.side_effect = [
+            {"results": {"mmlu": {"acc,none": 0.81}}},
+            {"results": {"mmlu": {"acc,none": 0.87}}},
+        ]
+
+        cli_main(["capability-check",
+                   "--abliterated", "fake/abl",
+                   "--stock", "fake/stock",
+                   "--device", "cpu",
+                   "--output-dir", str(tmp_path)])