def test_model_image(imsize, nproc, count, chunk=8):
psf = None
model_orig = create_model()
imfit = Imfit(model_orig,
psf=psf,
quiet=True,
nproc=nproc,
chunk_size=chunk)
shape = (imsize, imsize)
imfit.getModelImage(shape)
t1 = time.time()
imfit._modelObject._testCreateModelImage(count)
return time.time() - t1
def test_fitting():
psf = gaussian_psf(2.5, size=9)
model_orig = create_model()
imfit = Imfit(model_orig, psf=psf, quiet=False)
noise_level = 0.1
shape = (100, 100)
image = imfit.getModelImage(shape)
noise = image * noise_level
image += (np.random.random(shape) * noise)
mask = np.zeros_like(image, dtype='bool')
imfit.fit(image, noise, mask)
model_fitted = imfit.getModelDescription()
orig_params = get_model_param_array(model_orig)
fitted_params = get_model_param_array(model_fitted)
assert_allclose(orig_params, fitted_params, rtol=noise_level)
