Tạo fixtures với pytest
Khi xây dựng unit test, đôi khi bạn cần thiết lập một vài thứ trước khi bắt đầu kiểm thử. Nếu làm phần thiết lập này ngay trong unit test, bài kiểm thử sẽ khó đọc hơn và có thể phải lặp lại nhiều lần. May mắn là pytest cung cấp cách giải quyết bằng các fixtures.
Trong bài này, pandas đã được nhập với bí danh pd, và hàm extract() dưới đây đã sẵn sàng để bạn sử dụng!
def extract(file_path):
return pd.read_csv(file_path)
Bài tập này là một phần của khóa học
ETL và ELT với Python
Hướng dẫn bài tập
- Nhập thư viện
pytest. - Tạo một fixture
pytesttên làraw_tax_data. - Trả về DataFrame
raw_data.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____