Tworzenie fixture'ów za pomocą pytest
Podczas tworzenia testów jednostkowych czasem trzeba wykonać pewne przygotowania, zanim testowanie będzie mogło się rozpocząć. Robienie tego bezpośrednio w teście jednostkowym sprawia, że testy stają się trudniejsze do odczytania, a dodatkowo może wymagać powtarzania tego samego kodu wielokrotnie. Na szczęście pytest oferuje rozwiązanie tego problemu w postaci fixture'ów.
W tym ćwiczeniu pandas został już zaimportowany jako pd, a funkcja extract() pokazana poniżej jest gotowa 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 ____