test_ml_models.py

# test_ml_models.py
import pytest
from ml.predict_success import predict_token_success

def test_predict_token_success():
    fake_features = {
        'liquidity': 10000,
        'volume': 50000,
        'holder_growth': 1.05,
        'trading_velocity': 0.8
    }
    probability = predict_token_success(fake_features)
    assert 0.0 <= probability <= 1.0, "Predicted probability must be between 0 and 1"