def METHOD_NAME(self):
    num = self.imageModel.num_param_linear(
        self.kwargs_lens, self.kwargs_source, self.kwargs_lens_light, self.kwargs_ps
    )
    param = np.ones(num) * 10
    (
        kwargs_lens,
        kwargs_source,
        kwargs_lens_light,
        kwargs_ps,
    ) = self.imageModel.update_linear_kwargs(
        param,
        kwargs_lens=self.kwargs_lens,
        kwargs_source=self.kwargs_source,
        kwargs_lens_light=self.kwargs_lens_light,
        kwargs_ps=self.kwargs_ps,
    )
    assert kwargs_source[0]["amp"] == 10