diff --git a/tests/test_capability_check.py b/tests/test_capability_check.py index 0cf3b04..9ea88f4 100644 --- a/tests/test_capability_check.py +++ b/tests/test_capability_check.py @@ -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)])