def test_ex1d(self):
specrange = (0, self.nspec)
mask = np.zeros(self.image.shape, dtype=int)
ny = 20
flux, ivar = ex1d(self.image, mask, self.psf, yrange=[0,ny],
readnoise=1.0, specrange=specrange)
self.assertEqual(flux.shape, ivar.shape)
self.assertEqual(flux.shape[0], self.nspec)
self.assertEqual(flux.shape[1], ny)
def test_ex1d(self):
specrange = (0, self.nspec)
mask = np.zeros(self.image.shape, dtype=int)
ny = 20
flux, ivar = ex1d(self.image, mask, self.psf, yrange=[0,ny],
readnoise=1.0, specrange=specrange)
self.assertEqual(flux.shape, ivar.shape)
self.assertEqual(flux.shape[0], self.nspec)
self.assertEqual(flux.shape[1], ny)
def test_ex1d_model(self):
specrange = (2, self.nspec)
mask = np.zeros(self.image.shape, dtype=int)
ny = 20
flux, ivar, model = ex1d(self.image, mask, self.psf, yrange=[ny,2*ny],
readnoise=1.0, specrange=specrange, model=True)
self.assertEqual(flux.shape, ivar.shape)
self.assertEqual(flux.shape[0], self.nspec-2)
self.assertEqual(flux.shape[1], ny)
self.assertEqual(self.image.shape, model.shape)
def test_ex1d_model(self):
specrange = (2, self.nspec)
mask = np.zeros(self.image.shape, dtype=int)
ny = 20
flux, ivar, model = ex1d(self.image, mask, self.psf, yrange=[ny,2*ny],
readnoise=1.0, specrange=specrange, model=True)
self.assertEqual(flux.shape, ivar.shape)
self.assertEqual(flux.shape[0], self.nspec-2)
self.assertEqual(flux.shape[1], ny)
self.assertEqual(self.image.shape, model.shape)
