使用 assert 验证数据管道
要为数据管道编写单元测试,熟悉 assert 关键字和 isinstance() 函数非常重要。在本练习中,您将练习使用这两个工具来验证数据管道的组件。
函数 extract() 和 transform() 已为您提供,pandas 也已按 pd 导入。extract() 和 transform() 都会返回一个 DataFrame。祝您顺利完成!
本练习是课程的一部分
使用 Python 的 ETL 和 ELT
交互式实操练习
通过完成这段示例代码来试试这个练习。
raw_tax_data = extract("raw_tax_data.csv")
clean_tax_data = transform(raw_tax_data)
# Validate the number of columns in the DataFrame
____ len(clean_tax_data.columns) == ____