bf70ca3ac7
All 12 modules built per reviewed plan: - detector, state_machine (5-state phased FSM), canary, levels Phase B - notifier fanout (Discord + Telegram, bounded queue, retry, dead-letter) - audit (JSONL daily rotation), journal, report (weekly R-multiple PnL) - calibrate + labeler (Tk, lazy-imported), dryrun with acceptance gate - unified CLI: atm calibrate|label|dryrun|run|journal|report README + Phase 2 prop-firm TOS audit checklist included. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
69 lines
2.5 KiB
Python
69 lines
2.5 KiB
Python
"""Tests for atm.calibrate."""
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
|
|
def _minimal_config_data() -> dict:
|
|
return {
|
|
"window_title": "Test Chart",
|
|
"dot_roi": {"x": 0, "y": 0, "w": 100, "h": 100},
|
|
"chart_roi": {"x": 0, "y": 0, "w": 800, "h": 600},
|
|
"colors": {
|
|
"turquoise": {"rgb": [0, 200, 200], "tolerance": 30.0},
|
|
"yellow": {"rgb": [255, 255, 0], "tolerance": 30.0},
|
|
"dark_green": {"rgb": [0, 128, 0], "tolerance": 30.0},
|
|
"dark_red": {"rgb": [139, 0, 0], "tolerance": 30.0},
|
|
"light_green": {"rgb": [144, 238, 144], "tolerance": 30.0},
|
|
"light_red": {"rgb": [255, 182, 193], "tolerance": 30.0},
|
|
"gray": {"rgb": [128, 128, 128], "tolerance": 30.0},
|
|
},
|
|
"y_axis": {"p1_y": 100, "p1_price": 10000.0, "p2_y": 200, "p2_price": 9000.0},
|
|
"canary": {
|
|
"roi": {"x": 0, "y": 0, "w": 50, "h": 50},
|
|
"baseline_phash": "abc123",
|
|
"drift_threshold": 8,
|
|
},
|
|
"discord": {"webhook_url": "http://example.com/hook"},
|
|
"telegram": {"bot_token": "123:abc", "chat_id": "456"},
|
|
}
|
|
|
|
|
|
def test_write_config_and_marker(tmp_path: Path) -> None:
|
|
from atm.calibrate import write_config
|
|
from atm.config import Config
|
|
|
|
config_path = write_config(_minimal_config_data(), tmp_path)
|
|
|
|
assert config_path.exists()
|
|
assert config_path.suffix == ".toml"
|
|
|
|
# Must be loadable by Config.load
|
|
cfg = Config.load(config_path)
|
|
assert cfg.window_title == "Test Chart"
|
|
assert cfg.y_axis.p1_price == pytest.approx(10000.0)
|
|
|
|
# Marker must point at the filename (basename only)
|
|
marker = tmp_path / "current.txt"
|
|
assert marker.exists()
|
|
assert marker.read_text(encoding="utf-8").strip() == config_path.name
|
|
|
|
# Config.load_current should also work
|
|
cfg2 = Config.load_current(tmp_path)
|
|
assert cfg2.window_title == cfg.window_title
|
|
|
|
|
|
def test_import_safe() -> None:
|
|
"""Importing atm.calibrate must succeed in a headless environment (no tkinter at top-level)."""
|
|
import importlib # noqa: F401
|
|
import importlib.util
|
|
|
|
spec = importlib.util.find_spec("atm.calibrate")
|
|
assert spec is not None, "atm.calibrate module not found"
|
|
# Actually importing must not raise (tkinter is only used inside run_calibration)
|
|
mod = importlib.import_module("atm.calibrate")
|
|
assert hasattr(mod, "write_config")
|
|
assert hasattr(mod, "run_calibration")
|