def test_kk_3d():
x = np.linspace(-100, 100, 1000)
y = 2 / (2**2 + x**2)
y = y[:, None, None]
kkd = kkrelation(0 * y + 1, y, norm_by_bg=False)
assert_array_almost_equal(np.abs(kkd), np.sqrt(y))
def test_kk():
x = np.linspace(-100, 100, 1000)
y = 2 / (2**2 + x**2)
hilb_y_analytical = x / (2**2 + x**2)
kkd = kkrelation(0 * y + 1, np.exp(2 * y), pad_factor=10)
kkd_angle = np.angle(kkd)
assert_array_almost_equal(hilb_y_analytical, kkd_angle, decimal=4)
def test_kk_3d():
x = np.linspace(-100, 100, 1000)
y = 2/(2**2 + x**2)
y = y[:, None, None]
kkd = kkrelation(0*y + 1, y, norm_by_bg=False)
assert_array_almost_equal(np.abs(kkd), np.sqrt(y))
def test_kk():
x = np.linspace(-100, 100, 1000)
y = 2/(2**2 + x**2)
hilb_y_analytical = x/(2**2 + x**2)
kkd = kkrelation(0*y + 1, np.exp(2*y), pad_factor=10)
kkd_angle = np.angle(kkd)
assert_array_almost_equal(hilb_y_analytical, kkd_angle, decimal=4)
def test_kk_no_bg_norm():
x = np.linspace(-100, 100, 1000)
y = 2 / (2**2 + x**2)
kkd = kkrelation(0 * y + 1, y, norm_to_nrb=False)
assert_array_almost_equal(np.abs(kkd), np.sqrt(y))