[Kwant] Generalize current (and density) interpolation