def test_off_map(self) :
pixels = dirty_map.pixel_counts(self.data, self.ra_inds, self.dec_inds,
self.pix_counts, map_shape=(7,7))
self.assertTrue(not (8,1) in pixels)
self.assertTrue(not (9,5) in pixels)
self.assertTrue(not (8,4) in pixels)
self.assertTrue(not (7,3) in pixels)
self.assertTrue(sp.alltrue(self.pix_counts[pixels.index((5,2))] == 3))
def test_pix_counts_masked(self) :
self.data[2, 15] = ma.masked
self.data[4, 8] = ma.masked
pixels = dirty_map.pixel_counts(self.data, self.ra_inds, self.dec_inds,
self.pix_counts)
self.assertEqual(self.pix_counts[pixels.index((5,2)),15], 2)
self.assertEqual(self.pix_counts[pixels.index((5,2)),8], 3)
self.assertEqual(self.pix_counts[pixels.index((7,3)),8], 0)
def test_pix_counts_basic_cases(self) :
pixels = dirty_map.pixel_counts(self.data, self.ra_inds, self.dec_inds,
self.pix_counts)
self.assertTrue(sp.alltrue(self.pix_counts[pixels.index((8,1))] == 2))
self.assertTrue(sp.alltrue(self.pix_counts[pixels.index((5,2))] == 3))
self.assertTrue(sp.alltrue(self.pix_counts[pixels.index((9,5))] == 1))
self.assertTrue(sp.alltrue(self.pix_counts[pixels.index((7,3))] == 1))
self.assertTrue(sp.alltrue(self.pix_counts[pixels.index((8,4))] == 1))
def test_add_scan_noise_masked(self) :
self.data[2, 15] = ma.masked
self.data[4, 8] = ma.masked
pixels = dirty_map.pixel_counts(self.data, self.ra_inds, self.dec_inds,
self.pix_counts)
dirty_map.add_scan_noise(pixels, self.pix_counts, 2.0, self.noise_inv)
self.assertEqual(self.noise_inv[5,2,5,2,15], (2.0-4.0/9.0)/2.0)
self.assertEqual(self.noise_inv[5,2,5,2,4], (3.0-9.0/10.0)/2.0)
self.assertTrue(sp.allclose(self.noise_inv[:,:,7,3,8], 0.0))
self.assertEqual(self.noise_inv[5,2,7,3,15], (-2.0/9.0)/2.0)
def test_add_scan_noise(self) :
pixels = dirty_map.pixel_counts(self.data, self.ra_inds,
self.dec_inds, self.pix_counts)
dirty_map.add_scan_noise(pixels, self.pix_counts, 2.0, self.noise_inv)
self.assertTrue(sp.allclose(self.noise_inv[5,2,5,2,:],
(3.0-9.0/10)/2.0))
self.assertTrue(sp.allclose(self.noise_inv[5,3,5,3,:],
(1.0-1.0/10)/2.0))
self.assertTrue(sp.allclose(self.noise_inv[8,1,5,2,:],
(-6.0/10)/2.0))
self.assertTrue(sp.allclose(self.noise_inv[5,2,8,1,:],
(-6.0/10)/2.0))
def test_add_raises(self) :
pixels = dirty_map.pixel_counts(self.data, self.ra_inds, self.dec_inds,
self.pix_counts)
self.noise_inv = self.noise_inv[1,...]
self.assertRaises(ValueError, dirty_map.add_scan_noise, pixels,
self.pix_counts, 2.0, self.noise_inv)
