16 lines
571 B
Python
16 lines
571 B
Python
from pathlib import Path
|
|
|
|
from trading_bot.models.prediction import FEATURE_COLUMNS, predict_latest_probability
|
|
|
|
|
|
def test_predict_latest_probability_returns_probability_between_zero_and_one() -> None:
|
|
result = predict_latest_probability(
|
|
model_path=Path("notebooks/models/spy_xgb_v1.json"),
|
|
metadata_path=Path("notebooks/models/spy_xgb_v1_meta.json"),
|
|
raw_data_dir=Path("data/ibkr/daily"),
|
|
)
|
|
|
|
assert set(result.feature_columns).issubset(FEATURE_COLUMNS)
|
|
assert 0.0 <= result.probability <= 1.0
|
|
assert result.probability > 0.0
|