Refactor diffusion tests to be run with IOSim{POR}