Use LLMs to improve your code

pytest