Lagrangian modelling of multi-dimensional advection-diffusion with space-varying diffusivities: Theory and idealized test cases