Tworzenie fixtures z pytest
Pisząc testy jednostkowe, czasem trzeba wykonać pewne czynności przygotowawcze, zanim samo testowanie będzie mogło się rozpocząć. Umieszczanie tej konfiguracji wewnątrz testu jednostkowego utrudnia jego czytelność, a poza tym często wymaga wielokrotnego powtarzania tego samego kodu. Na szczęście pytest rozwiązuje ten problem za pomocą fixtures.
W tym ćwiczeniu pandas zostało zaimportowane jako pd, a poniższa funkcja extract() jest dostępna do użycia!
def extract(file_path):
return pd.read_csv(file_path)
To ćwiczenie jest częścią kursu
ETL i ELT w Pythonie
Instrukcje do ćwiczenia
- Zaimportuj bibliotekę
pytest. - Utwórz fixture
pytesto nazwieraw_tax_data. - Zwróć DataFrame
raw_data.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Import pytest
____
# Create a pytest fixture
@pytest.fixture()
def ____():
raw_data = extract("raw_tax_data.csv")
# Return the raw DataFrame
return ____