| 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 | |