Would recommend to try out anthropic sonnet 3.5 for this one - usually generates decent unit tests for reasonably sized functions