Criando acessórios com o pytest
Ao criar testes unitários, às vezes você precisará fazer algumas configurações antes de iniciar o teste. Fazer essa configuração em um teste de unidade pode dificultar a leitura dos testes e talvez você precise repeti-los várias vezes. Felizmente, o site pytest oferece uma maneira de resolver esses problemas, com acessórios.
Para este exercício, pandas 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 exercício faz parte do curso
ETL e ELT em Python
Instruções de exercício
- Importe a biblioteca
pytest. - Crie um acessório
pytestchamadoraw_tax_data. - Retorna o
raw_dataDataFrame.
Exercício interativo prático
Experimente este exercício preenchendo 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 ____