import dataframely as dy
import polars as pl

from tests.test_validate_date_col import DateSchema


class PackagesSchema(DateSchema):
    package: dy.String = dy.String(nullable=False)
    version: dy.String = dy.String(nullable=False)
    count: dy.Integer = dy.Integer(nullable=False)


def test_packages_schema():
    _ = PackagesSchema.validate(
        pl.scan_csv(
            "output/packages.csv",
            schema={
                "date": pl.Date,
                "package": pl.String,
                "version": pl.String,
                "count": pl.UInt16,
            },
        ).collect(engine="streaming")
    )