pytest로 픽스처(fixture) 만들기
단위 테스트를 작성하다 보면 테스트를 시작하기 전에 약간의 준비 작업이 필요할 때가 있습니다. 이런 준비 작업을 단위 테스트 안에서 직접 처리하면 테스트 코드를 읽기 어려워지고, 같은 작업을 여러 번 반복해야 할 수도 있습니다. 다행히 pytest는 픽스처(fixture)를 통해 이러한 문제를 해결할 수 있는 방법을 제공합니다.
이번 연습 문제에서는 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 ____