From 1401864a7ffd0dff7083e6005aed161641e8ef92 Mon Sep 17 00:00:00 2001 From: Jeremy Kubica <104161096+jeremykubica@users.noreply.github.com> Date: Fri, 22 Sep 2023 11:39:48 -0400 Subject: [PATCH] Bug fixes --- tests/test_layered_image.py | 12 ++++++------ tests/test_psf.py | 6 +++--- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/tests/test_layered_image.py b/tests/test_layered_image.py index 5d7cf43c..334233e4 100644 --- a/tests/test_layered_image.py +++ b/tests/test_layered_image.py @@ -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)) diff --git a/tests/test_psf.py b/tests/test_psf.py index 01e2aee0..4d088c90 100644 --- a/tests/test_psf.py +++ b/tests/test_psf.py @@ -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)