Add tests

Author: benoit-cty

tests/cli/test_cli_main.py

@@ -324,6 +324,24 @@ def fake_run_and_monitor(ctx, offline=False, **kwargs):
     assert captured["kwargs"]["country_iso_code"] == "FRA"
 
 
+def test_monitor_delegates_online_mode_to_run_and_monitor(monkeypatch):
+    captured = {}
+
+    def fake_run_and_monitor(ctx, offline=False, **kwargs):
+        captured["offline"] = offline
+        captured["kwargs"] = kwargs
+        return "ok"
+
+    monkeypatch.setattr(cli_main, "run_and_monitor", fake_run_and_monitor)
+    monkeypatch.setattr(cli_main, "get_existing_exp_id", lambda: "exp-1")
+
+    ctx = SimpleNamespace(args=["python", "train.py"])
+    result = cli_main.monitor(ctx=ctx, api=True)
+    assert result == "ok"
+    assert captured["offline"] is False
+    assert captured["kwargs"]["save_to_api"] is True
+
+
 def test_monitor_delegates_to_run_and_monitor_with_extra_args(monkeypatch):
     captured = {}
 
@@ -340,3 +358,21 @@ def fake_run_and_monitor(ctx, **kwargs):
     assert result == "ok"
     assert captured["args"] == ["python", "train.py"]
     assert captured["kwargs"]["save_to_api"] is False
+
+
+def test_monitor_no_api_skips_experiment_id_requirement(monkeypatch):
+    captured = {}
+
+    def fake_run_and_monitor(ctx, offline=False, **kwargs):
+        captured["offline"] = offline
+        captured["kwargs"] = kwargs
+        return "ok"
+
+    monkeypatch.setattr(cli_main, "run_and_monitor", fake_run_and_monitor)
+    monkeypatch.setattr(cli_main, "get_existing_exp_id", lambda: None)
+
+    ctx = SimpleNamespace(args=["python", "train.py"])
+    result = cli_main.monitor(ctx=ctx, api=False)
+    assert result == "ok"
+    assert captured["offline"] is False
+    assert captured["kwargs"]["save_to_api"] is False

tests/cli/test_monitor.py

@@ -91,6 +91,37 @@ def wait(self):
     assert captured["kwargs"]["country_iso_code"] == "FRA"
 
 
+def test_run_and_monitor_uses_online_tracker_by_default(monkeypatch):
+    captured = {}
+
+    class FakeOnlineTracker(FakeTracker):
+        def __init__(self, **kwargs):
+            captured["kwargs"] = kwargs
+            super().__init__()
+
+    class FakePopen:
+        def __init__(self, command, text=True):
+            pass
+
+        def wait(self):
+            return 0
+
+    monkeypatch.setattr(monitor_module, "EmissionsTracker", FakeOnlineTracker)
+    monkeypatch.setattr(monitor_module, "OfflineEmissionsTracker", FakeTracker)
+    monkeypatch.setattr(monitor_module.subprocess, "Popen", FakePopen)
+    monkeypatch.setattr(monitor_module, "print", lambda *args, **kwargs: None)
+
+    with pytest.raises(typer.Exit) as exc_info:
+        monitor_module.run_and_monitor(
+            SimpleNamespace(args=["echo", "hi"]),
+            save_to_api=True,
+        )
+
+    assert exc_info.value.exit_code == 0
+    assert captured["kwargs"]["tracking_mode"] == "process"
+    assert captured["kwargs"]["save_to_api"] is True
+
+
 def test_run_and_monitor_handles_keyboard_interrupt(monkeypatch):
     process_info = {"terminated": 0, "killed": 0}