测试模式(schema)
团队已经修正了 expected DataFrame 中的取值,但 Polars 将 total_checkouts 推断为 64 位,而流水线产出的是 32 位。系统已经为您打印了两个模式,便于查看不一致之处。请修正期望的 dtype,然后断言两个模式相同。
actual、expected 和 assert_schema_equal 已预加载。
本练习是课程的一部分
使用 Polars 扩展与优化数据流水线
练习说明
- 将
expected中的total_checkouts转换为pl.Int32。 - 断言
actual与expected的模式相等。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Match the dtype the pipeline produces
expected = expected.with_columns(
pl.col("total_checkouts").cast(pl.____)
)
# Schema-level assertion
____(actual.schema, expected.schema)
print("Schema assertion passed.")
print(actual.schema)