Checkpoint experiments through SWE pilot v2

tests/test_run_board.py

@@ -0,0 +1,83 @@
+from pathlib import Path
+import runpy
+from unittest.mock import patch
+
+import pytest
+
+SCRIPT = Path(__file__).parents[1] / 'scripts/run_board.py'
+REVISION = 'a' * 40
+
+
+def test_interactive_defaults_preserve_matched_pilot():
+    module = runpy.run_path(str(SCRIPT))
+    values = iter(['', REVISION, '', '', '', '', '', ''] + [''] * 12)
+    args = module['interactive_arguments'](lambda _: next(values))
+    config = dict(zip(args[::2], args[1::2]))
+    assert config['--model'] == 'glm'
+    assert config['--dataset-revision'] == REVISION
+    assert config['--agents-per-cohort'] == '2'
+    assert config['--cohorts'] == '2'
+    assert config['--teams'] == '2'
+    assert config['--sampling'] == 'balanced-repeat'
+    assert config['--prompt-variant'] == 'D'
+    assert '--execute' not in args
+
+
+def test_different_population_prompts_sampling_and_contributor():
+    module = runpy.run_path(str(SCRIPT))
+    values = iter(['muse', REVISION, '', '', '', '', '', '', '3', '2', '3'] + [''] * 9)
+    args = module['interactive_arguments'](lambda _: next(values))
+    config = dict(zip(args[::2], args[1::2]))
+    assert config['--model'] == 'muse'
+    assert config['--sampling'] == 'balanced-repeat'
+    assert config['--teams'] == '3'
+
+
+def test_invalid_number_reprompts():
+    module = runpy.run_path(str(SCRIPT))
+    values = iter(['0', '-1', 'abc', '2'])
+    assert module['ask']('Agents', 3, module['positive'],
+                         input_fn=lambda _: next(values)) == '2'
+
+
+def test_forwarding_keeps_paths_and_model_literal_and_preview_default():
+    module = runpy.run_path(str(SCRIPT))
+    with patch('subprocess.run') as run:
+        run.return_value.returncode = 0
+        assert module['main'](['--model', 'openrouter/vendor/model',
+                              '--out', 'logs/path with spaces']) == 0
+    argv = run.call_args.args[0]
+    assert argv[-1] == 'logs/path with spaces'
+    assert '--execute' not in argv
+    assert run.call_args.kwargs.get('shell', False) is False
+
+
+def test_execute_is_explicit_and_child_failure_propagates():
+    module = runpy.run_path(str(SCRIPT))
+    with patch('subprocess.run') as run:
+        run.return_value.returncode = 7
+        assert module['main'](['--execute']) == 7
+    assert run.call_args.args[0][-1] == '--execute'
+
+
+def test_cancelled_interactive_never_launches():
+    with patch('builtins.input', side_effect=EOFError), patch('subprocess.run') as run:
+        module = runpy.run_path(str(SCRIPT))
+        assert module['main'](['--interactive', '--execute']) == 130
+        run.assert_not_called()
+
+
+def test_preview_cannot_be_overridden_to_spend():
+    module = runpy.run_path(str(SCRIPT))
+    with patch('subprocess.run') as run, pytest.raises(SystemExit):
+        module['main'](['--preview', '--execute'])
+    run.assert_not_called()
+
+
+def test_child_interrupt_does_not_claim_no_run_started(capsys):
+    module = runpy.run_path(str(SCRIPT))
+    with patch('subprocess.run', side_effect=KeyboardInterrupt):
+        assert module['main'](['--execute']) == 130
+    message = capsys.readouterr().err
+    assert 'Runner interrupted' in message
+    assert 'no experiment started' not in message