Diffusion Data Augmentation Python Code Tester Step