I already use gpt4 for this. Works quite well if you show it a couple of reference unit tests.