import pandas as pd

from tools.run_mlp_sopr_entry_ablation import calibrate_sopr_cutoff


def test_sopr_cutoff_does_not_read_future_rows():
    data = pd.DataFrame({
        "time": pd.to_datetime(["2017-12-01", "2017-12-02", "2017-12-03", "2018-01-01"]),
        "sopr_norm": [-0.8, -0.4, 0.2, -1.0],
    })

    cutoff = calibrate_sopr_cutoff(data, "2017-12-03", 0.10)

    assert cutoff == -0.72