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 ____"