Membuat fixture dengan pytest

Saat membangun unit test, terkadang Anda perlu melakukan sedikit penyiapan sebelum pengujian dapat dimulai. Melakukan penyiapan ini di dalam unit test dapat membuat pengujian lebih 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() berikut 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