Pytest Test Class Example