如何使用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