wehub-resource-sync
99 lines
3.2 KiB
Python
99 lines
3.2 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from tests.pm_helpers import (
|
|
RECORD_REPLAY,
|
|
RECORD_REWRITE,
|
|
TIER_ON_DEMAND,
|
|
TIER_PER_COMMIT,
|
|
TIER_WEEKLY,
|
|
collect_chapter_notebooks,
|
|
current_test_tier,
|
|
get_record_mode,
|
|
get_reruns,
|
|
get_tier,
|
|
)
|
|
|
|
|
|
def test_collect_chapter_notebooks_keeps_real_notebooks_and_skips_helpers() -> None:
|
|
notebooks = {path.as_posix() for path in collect_chapter_notebooks(Path("."), range(1, 28))}
|
|
|
|
assert "06_strategy_definition/02_case_study_overview.py" in notebooks
|
|
assert "08_financial_features/case_study_feature_summary.py" in notebooks
|
|
assert "11_ml_pipeline/08_ml_backtest_intro.py" in notebooks
|
|
assert "16_strategy_simulation/01_backtest_first_principles.py" in notebooks
|
|
assert "21_rl_execution_hedging/07_backtest_with_impact.py" in notebooks
|
|
|
|
assert "03_market_microstructure/filter_itch_symbol.py" not in notebooks
|
|
assert "03_market_microstructure/lob_utils.py" not in notebooks
|
|
assert "03_market_microstructure/lob_generator.py" not in notebooks
|
|
assert "13_dl_time_series/dl_utils.py" not in notebooks
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Tier / reruns / record_mode helpers
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_get_tier_defaults_to_per_commit() -> None:
|
|
assert get_tier({}) == TIER_PER_COMMIT
|
|
assert get_tier({"tier": None}) == TIER_PER_COMMIT
|
|
|
|
|
|
def test_get_tier_accepts_valid_values() -> None:
|
|
assert get_tier({"tier": "per_commit"}) == TIER_PER_COMMIT
|
|
assert get_tier({"tier": "weekly"}) == TIER_WEEKLY
|
|
assert get_tier({"tier": "on_demand"}) == TIER_ON_DEMAND
|
|
|
|
|
|
def test_get_tier_rejects_invalid() -> None:
|
|
with pytest.raises(ValueError, match="Invalid tier"):
|
|
get_tier({"tier": "nightly"})
|
|
|
|
|
|
def test_current_test_tier_defaults_to_per_commit(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.delenv("ML4T_TEST_TIER", raising=False)
|
|
assert current_test_tier() == TIER_PER_COMMIT
|
|
|
|
|
|
def test_current_test_tier_reads_env(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("ML4T_TEST_TIER", "weekly")
|
|
assert current_test_tier() == TIER_WEEKLY
|
|
monkeypatch.setenv("ML4T_TEST_TIER", "on_demand")
|
|
assert current_test_tier() == TIER_ON_DEMAND
|
|
|
|
|
|
def test_current_test_tier_rejects_invalid(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("ML4T_TEST_TIER", "bogus")
|
|
with pytest.raises(ValueError, match="Invalid ML4T_TEST_TIER"):
|
|
current_test_tier()
|
|
|
|
|
|
def test_get_reruns_default_zero() -> None:
|
|
assert get_reruns({}) == 0
|
|
|
|
|
|
def test_get_reruns_returns_int() -> None:
|
|
assert get_reruns({"reruns": 3}) == 3
|
|
|
|
|
|
def test_get_reruns_rejects_negative_or_nonint() -> None:
|
|
with pytest.raises(ValueError):
|
|
get_reruns({"reruns": -1})
|
|
with pytest.raises(ValueError):
|
|
get_reruns({"reruns": "2"})
|
|
|
|
|
|
def test_get_record_mode_defaults_to_replay() -> None:
|
|
assert get_record_mode({}) == RECORD_REPLAY
|
|
|
|
|
|
def test_get_record_mode_accepts_rewrite() -> None:
|
|
assert get_record_mode({"record_mode": "rewrite"}) == RECORD_REWRITE
|
|
|
|
|
|
def test_get_record_mode_rejects_invalid() -> None:
|
|
with pytest.raises(ValueError, match="Invalid record_mode"):
|
|
get_record_mode({"record_mode": "none"})
|