24 lines
631 B
Python
24 lines
631 B
Python
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")
|
|
)
|