def test_gzz(): "gravmag.prism.gzz python vs cython implementation" py = _prism.gzz(xp, yp, zp, model) cy = _cprism.gzz(xp, yp, zp, model) diff = np.abs(py - cy) assert np.all(diff <= precision), 'max diff: %g' % (max(diff))
def test_gzz_ne(): "gravmag.prism.gzz python vs numexpr implementation" py = _prism.gzz(xp, yp, zp, model) ne = _neprism.gzz(xp, yp, zp, model) diff = np.abs(py - ne) assert np.all(diff <= precision), 'max diff: %g' % (max(diff))