Test diffusion implementation