Crear fixtures con pytest
Al crear pruebas unitarias, a veces tendrás que hacer algo de preparación antes de empezar a probar. Hacer esta preparación dentro de una prueba unitaria puede dificultar la lectura de las pruebas y puede que tengas que repetirla varias veces. Por suerte, pytest ofrece una forma de resolver estos problemas con las fixtures.
En este ejercicio, pandas se ha importado como pd, y la función extract() que ves a continuación está disponible para usarla.
def extract(file_path):
return pd.read_csv(file_path)
Este ejercicio forma parte del curso
ETL and ELT con Python
Instrucciones del ejercicio
- Importa la biblioteca
pytest. - Crea una fixture de
pytestllamadaraw_tax_data. - Devuelve el DataFrame
raw_data.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____