from training.eval import generate_report
from training.heuristic_baselines import POLICIES


def test_all_policies_present():
    assert {"fifo", "earliest_deadline_first", "best_fit_vram"} <= set(POLICIES)


def test_evaluation_report_has_expected_rows():
    report = generate_report()
    assert len(report) == 12
    assert all("total_reward" in row for row in report)