python - 如何对使用“pandas.read_sql()”读取数据的函数有效的列名进行单元测试？

Question

我有一个从 MySQL 数据库读取数据的功能：

def example(mysql_engine) -> DataFrame:
    query = """SELECT col_1 FROM xxx.xxx"""
    df = pandas.read_sql(query, mysql_engine)

    return df

mysql_engine由另一个函数返回，现在我想编写单元测试来验证 中的列名col_1和数据类型df，有没有我可以遵循的示例？我需要在单元测试中设置一个真正的数据库引擎吗？

Answer 1

您可以assert col_1 in df.columns用于单元测试。

对于数据库引擎，您可以使用模拟对象。您可以从标准库中检查此链接。