Membuat fixture dengan pytest

Saat membangun unit test, kadang Anda perlu melakukan sedikit penyiapan sebelum pengujian dapat dimulai. Melakukan penyiapan ini di dalam unit test dapat membuat pengujian sulit dibaca, dan mungkin harus diulang beberapa kali. Untungnya, pytest menyediakan cara untuk mengatasi masalah ini melalui fixture.

Untuk latihan ini, pandas telah diimpor sebagai pd, dan fungsi extract() di bawah ini tersedia untuk digunakan!

def extract(file_path):
    return pd.read_csv(file_path)

Latihan ini adalah bagian dari kursus

ETL dan ELT di Python

Lihat Kursus

Petunjuk latihan

Impor pustaka pytest.
Buat fixture pytest bernama raw_tax_data.
Kembalikan DataFrame raw_data.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Import pytest
____

# Create a pytest fixture
@pytest.fixture()
def ____():
	raw_data = extract("raw_tax_data.csv")
    
    # Return the raw DataFrame
	return ____

Edit dan Jalankan Kode