File size: 650 Bytes
b67668b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | from src.agent.schemas import ExtractedData
def test_schema_accepts_valid_employee():
payload = {
"employees": [
{
"user_id": 101,
"name": "Michael Chen",
"age": 29,
"email": None,
"salary": 120000,
"join_date": "2024-01-01",
"department": "Artificial Intelligence",
"performance_score": 9.5,
"location": "Chicago",
"job_title": "Engineer",
}
]
}
data = ExtractedData.model_validate(payload)
assert data.employees[0].user_id == 101
|