การสร้าง fixture ด้วย pytest

เมื่อสร้าง unit test บางครั้งต้องเตรียมการบางอย่างก่อนเริ่มทดสอบ การเตรียมการนี้ภายใน unit test อาจทำให้เทสต์อ่านยากขึ้น และอาจต้องทำซ้ำหลายครั้ง โชคดีที่ pytest มีวิธีแก้ปัญหานี้ด้วย fixture

สำหรับแบบฝึกหัดนี้ pandas ถูก import มาเป็น pd แล้ว และฟังก์ชัน extract() ด้านล่างพร้อมให้ใช้งาน!

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

แบบฝึกหัดนี้เป็นส่วนหนึ่งของหลักสูตร

ETL and ELT ใน Python

ดูคอร์ส

คำแนะนำการฝึกหัด

Import ไลบรารี pytest
สร้าง fixture ของ pytest ชื่อ raw_tax_data
ส่งคืน DataFrame ชื่อ raw_data

แบบฝึกหัดเชิงโต้ตอบแบบลงมือทำ

ลองทำแบบฝึกหัดนี้โดยเติมโค้ดตัวอย่างนี้ให้สมบูรณ์

# Import pytest
____

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

แก้ไขและรันโค้ด