Fix `abs` for DistArrays and LocalArrays

and add some tests.