evals/results/workspaces/2026-08-26-05-gate-economy/tests/test_records.py

import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parent.parent))

from src.records.model import Record, by_team, total_size

RECORDS = [
    Record(id="a1", team="alpha", name="one", size_bytes=100),
    Record(id="b2", team="beta", name="two", size_bytes=250),
]


def test_total_size():
    assert total_size(RECORDS) == 350


def test_by_team():
    assert [record.id for record in by_team(RECORDS, "alpha")] == ["a1"]