22 lines
544 B
Python
22 lines
544 B
Python
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")
|
|
)
|