Criando fixtures com pytest
Ao criar testes de unidade, às vezes você vai precisar fazer uma preparação antes de começar a testar. Fazer essa preparação dentro do próprio teste pode deixar o código mais difícil de ler e pode ter que ser repetida várias vezes. Felizmente, o pytest oferece uma forma de resolver esses problemas com as fixtures.
Neste exercício, pandas já foi importado como pd, e a função extract() mostrada abaixo está disponível para uso!
def extract(file_path):
return pd.read_csv(file_path)
Este exercicio faz parte do curso
ETL e ELT em Python
Instruções do exercicio
- Importe a biblioteca
pytest. - Crie uma fixture do
pytestchamadaraw_tax_data. - Retorne o DataFrame
raw_data.
exercicio interativo prático
Tente este exercicio completando este código de exemplo.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____