我可以将 dataframe 传递给@pytest.mark.parametrize 吗？

Question

我想将 dataframe 传递给@pytest.mark.parametrize 。 数据帧存储在conftest.py上。 不使用引用数据帧的@pytest.mark.parametrize的单元测试成功执行。

但是，当我应用@pytest.mark.parameterize时，代码返回TypeError: 'function' object is not subscriptable

数据框在conftest.py中作为函数开发。 例如：

@pytest.fixture(scope="module")
def df_vartypes():
    data = {
        "Name": ["tom", "nick", "krish", "jack"],
        "City": ["London", "Manchester", "Liverpool", "Bristol"],
        "Age": [20, 21, 19, 18],
        "Marks": [0.9, 0.8, 0.7, 0.6],
        "dob": pd.date_range("2020-02-24", periods=4, freq="T"),
    }

    df = pd.DataFrame(data)

    return df

单元测试：

_cat_num_vars = [
    (df_enc, "var_A", ["var_A"], []),
    (df_enc_numeric, "var_B", [], ["var_B"]),
    # TODO: Datetime test
    (df_vartypes, None, ["Name", "City"], ["Age", "Marks"]),
    (df_enc_numeric, None, [], ["var_A", "var_B", "target"])
]

@pytest.mark.parametrize(
    "_df, _variables, _categorical_vars, _numerical_vars", _cat_num_vars
)
def test_find_categorical_and_numeric_vars_pass_diff_var_permutations(
        _df, _variables, _categorical_vars, _numerical_vars
):
    assert (_find_categorical_and_numerical_variables(
        _df, _variables) == (_categorical_vars, _numerical_vars)
            )