Adding assert to check if dx/dz are equal in whole domain