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 merupakan bagian dari kursus

ETL dan ELT di Python

Lihat Kursus

Instruksi latihan

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

Latihan interaktif langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

# 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