如何使用pytest生成，读取和删除临时csv文件？

Question

我想测试一个处理csv文件的函数。

如何使用pytest生成文件（带有指定内容），对该文件运行测试，然后再次删除文件？

Answer 1

我用了一个夹具。它似乎有效。

import pandas as pd
import pytest

@pytest.fixture(scope='session')
def csv_file(tmpdir_factory):
    N = 10
    index = range(N)
    even = [(n % 2 == 0) for n in range(N)]
    dataframe = pd.DataFrame({'even': even}, index=index) 
    filename = str(tmpdir_factory.mktemp('data').join('data.csv'))
    dataframe.to_csv(filename)
    return filename


def some_test(csv_file):
    pass