Skip to content

Commit

Permalink
Bug fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
jeremykubica committed Sep 22, 2023
1 parent b62c5ad commit 1401864
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 9 deletions.
12 changes: 6 additions & 6 deletions tests/test_layered_image.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,23 +79,23 @@ def test_convolve_psf(self):
msk0 = self.image.get_mask()

# Create a copy of the image.
img_b = layered_image(sci, var, mask, self.p)
img_b = layered_image(sci0, var0, msk0, self.p)

# A no-op PSF does not change the image.
img_b.convolve_given_psf(psf())
sci1 = img_b.get_science()
var1 = img_b.get_variance()
for y in range(img2.get_height()):
for x in range(img2.get_width()):
for y in range(img_b.get_height()):
for x in range(img_b.get_width()):
self.assertAlmostEqual(sci0.get_pixel(x, y), sci1.get_pixel(x, y))
self.assertAlmostEqual(var0.get_pixel(x, y), var1.get_pixel(x, y))

# The default PSF (stdev=1.0) DOES have the image.
img_b.convolve_psf(psf())
img_b.convolve_psf()
sci1 = img_b.get_science()
var1 = img_b.get_variance()
for y in range(img2.get_height()):
for x in range(img2.get_width()):
for y in range(img_b.get_height()):
for x in range(img_b.get_width()):
self.assertNotAlmostEqual(sci0.get_pixel(x, y), sci1.get_pixel(x, y))
self.assertNotAlmostEqual(var0.get_pixel(x, y), var1.get_pixel(x, y))

Expand Down
6 changes: 3 additions & 3 deletions tests/test_psf.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,9 @@ def setUp(self):

def test_make_noop(self):
psf0 = psf()
self.assertEqual(psf1.get_size(), 1)
self.assertEqual(psf1.get_dim(), 1)
self.assertEqual(psf1.get_radius(), 0)
self.assertEqual(psf0.get_size(), 1)
self.assertEqual(psf0.get_dim(), 1)
self.assertEqual(psf0.get_radius(), 0)

kernel0 = psf0.get_kernel()
self.assertEqual(len(kernel0), 1)
Expand Down

0 comments on commit 1401864

Please sign in to comment.