Crear fixtures con pytest
Cuando construyas pruebas unitarias, a veces tendrás que hacer un poco de configuración antes de que puedan comenzar las pruebas. Hacer esta configuración dentro de una prueba unitaria puede hacer que las pruebas sean más difíciles de leer, y puede que haya que repetirlas varias veces. Por suerte, pytest ofrece una forma de resolver estos problemas, con accesorios.
Para este ejercicio, pandas se ha importado como pd, ¡y la función extract() que se muestra a continuación está disponible para su uso!
def extract(file_path):
return pd.read_csv(file_path)
Este ejercicio forma parte del curso
Introducción a las canalizaciones de datos
Instrucciones de ejercicio
- Importa la biblioteca
pytest. - Crea una fijación
pytestllamadaraw_tax_data. - Devuelve el DataFrame
raw_data.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este 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 ____