def test_empty(self):
z_source = 1.5
lens_model_list = []
redshift_list = []
lensModelMutli = MultiPlane(z_source=z_source,
lens_model_list=lens_model_list,
lens_redshift_list=redshift_list)
kwargs_lens = []
f_xx_multi, f_xy_multi, f_yx_multi, f_yy_multi = lensModelMutli.hessian(
1, 0, kwargs_lens, diff=0.000001)
npt.assert_almost_equal(0, f_xx_multi, decimal=5)
npt.assert_almost_equal(0, f_xy_multi, decimal=5)
npt.assert_almost_equal(0, f_yx_multi, decimal=5)
npt.assert_almost_equal(0, f_yy_multi, decimal=5)
def test_sis_hessian(self):
z_source = 1.5
lens_model_list = ['SIS']
redshift_list = [0.5]
lensModelMutli = MultiPlane(z_source=z_source,
lens_model_list=lens_model_list,
lens_redshift_list=redshift_list)
lensModel = LensModel(lens_model_list=lens_model_list)
kwargs_lens = [{'theta_E': 1, 'center_x': 0, 'center_y': 0}]
f_xx_simple, f_xy_simple, f_yx_simple, f_yy_simple = lensModel.hessian(
1, 0, kwargs_lens)
f_xx_multi, f_xy_multi, f_yx_multi, f_yy_multi = lensModelMutli.hessian(
1, 0, kwargs_lens, diff=0.000001)
npt.assert_almost_equal(f_xx_simple, f_xx_multi, decimal=5)
npt.assert_almost_equal(f_xy_simple, f_xy_multi, decimal=5)
npt.assert_almost_equal(f_yx_simple, f_yx_multi, decimal=5)
npt.assert_almost_equal(f_yy_simple, f_yy_multi, decimal=5)