[code_witch_sam]

why not have a system to write the tests as well?

[emoII]

Won't the prompt itself be the test? TDD is literally just providing an expected output on a given input, before implementing the functionality. The prompt we give AIs follows the same pattern, "I have a data that looks like ___, visualize it in a graph that ____"