def test_measure_psf(self, case):
sigma, aspect_ratio = case
psf = PSF()
psf.generate(sigma=sigma, aspect_ratio=aspect_ratio)
sigmas = psf.measure_psf_sigma()
self.assertEqual(len(sigmas), 3)
self.assertAlmostEqual(sigma/sigmas[1], 1, 1)
self.assertAlmostEqual(sigma/sigmas[2], 1, 1)
self.assertAlmostEqual(sigma*aspect_ratio/sigmas[0], 1, 1)
def test_convolve(self):
img = Image()
arr = np.zeros([50, 50, 50])
arr[10:-10, 10:-10, 10:-10] = 255
img.image = arr
psf = PSF()
psf.generate(sigma=5, aspect_ratio=4)
img.convolve(psf)
self.assertEqual(len(img.image.shape), len(arr.shape))
self.assertEqual(img.metadata['Convolved'], True)
def test_convolve_None(self):
img = Image()
psf = PSF()
psf.generate(sigma=5, aspect_ratio=4)
self.assertRaises(ValueError, img.convolve, psf)