pytest로 픽스처 만들기
유닛 테스트를 작성할 때는 테스트를 시작하기 전에 약간의 사전 준비가 필요한 경우가 있습니다. 이 준비 작업을 유닛 테스트 내부에서 처리하면 테스트를 읽기 어렵게 만들고, 여러 번 반복해야 할 수도 있습니다. 다행히 pytest에는 이러한 문제를 해결할 수 있는 픽스처 기능이 있습니다.
이번 연습에서는 pandas가 pd로 임포트되어 있으며, 아래에 보이는 extract() 함수를 사용할 수 있습니다!
def extract(file_path):
return pd.read_csv(file_path)
이 연습은 강의의 일부입니다
Python으로 ETL과 ELT
연습 안내
pytest라이브러리를 임포트하세요.raw_tax_data라는 이름의pytest픽스처를 만드세요.raw_dataDataFrame을 반환하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____