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)