import dataframely as dy
import polars as pl

from tests.test_validate_date_col import DateSchema


class KernelsSchema(DateSchema):
    kernel: dy.String = dy.String(nullable=False)
    downloads: dy.Integer = dy.Integer(nullable=False)


def test_kernels_schema():
    _ = KernelsSchema.validate(
        pl.scan_csv(
            "output/kernels.csv",
            schema={
                "date": pl.Date,
                "kernel": pl.String,
                "downloads": pl.UInt32,
            },
        ).collect(engine="streaming")
    )