Tạo fixtures với pytest
Khi xây dựng unit test, đôi lúc bạn cần làm một chút thiết lập trước khi bắt đầu kiểm thử. Nếu thực hiện phần thiết lập này ngay trong unit test, bài test 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 những vấn đề này bằng fixtures.
Trong bài tập này, pandas đã được import với tên 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
- Import thư viện
pytest. - Tạo một
pytestfixture tê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 ____