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))
