Python read_text() 添加额外的字符串

Question

我创建了一个函数，该函数接收带有动态 SQL（和参数）的文件，然后返回最终的 SQL 结果。 它似乎添加了额外的字符串，因为我的单元测试失败并且它显示了额外的字符串。

我的功能：

def file_to_sql(filename, kwargs=None):
    """Read file and return SQL statement.

    Parameters
    ----------
    filename: str
        Name of the file to read.
    kwargs : dict
        Keyword arguments to be passed into your SQL statement.
    """

    python_dir = Path('logic/extract/python3')
    python_file_path = python_dir / '{}'.format(filename)
    sql = python_file_path.read_text().format(**kwargs)

return sql

我的单元测试：

def test_python_file_with_kwargs_to_sql():
    sql = file_to_sql(filename="select_all_from_{table}.py", kwargs={
        'table': 'sandbox.test_table'
        })

    assert sql == "SELECT * FROM sandbox.test_table;"

select_all_from_{table}.py:

"SELECT * FROM {table};"

单元测试失败消息：

================================== FAILURES ===================================
_____________________ test_python_file_with_kwargs_to_sql _____________________

    def test_python_file_with_kwargs_to_sql():
        sql = file_to_sql(filename="select_all_from_{table}.py", kwargs={
            'table': 'sandbox.test_table'
            })

>       assert sql == "SELECT * FROM sandbox.test_table;"
E       assert '"SELECT * FR....test_table;"' == 'select * from...x.test_table;'
E         - "SELECT * FROM sandbox.test_table;"
E         + SELECT * FROM sandbox.test_table;

tests\unit_tests\transform\test_preprocess.py:19: AssertionError
================ 1 failed, 2 passed, 4 skipped in 1.40 seconds ================

Answer 1

您的单元测试失败有两个原因：

1. 您的sql字符串是大小写混合的（“SELECT”和“FROM”是大写），而您的比较字符串是全部小写。

2. select_all_from_{table}.py中的文本是双引号，但您的比较字符串没有双引号。

由于read_text()将文件内容读取为str您不需要双引号。 从您的文件中删除双引号，为了好的做法，请进行不区分大小写的测试：

sql.lower() == 'select * from sandbox.test_table'.lower()

或者只是sql.lower() == 'select * from sandbox.test_table'在这种特殊情况下就足够了。

对于单元测试，最好对输入和输出保持警惕，并确保测试准确无误。