def test_invalid_uncert_unit(self): ccd = self.ccd # we must do this to avoid ccddata self protection uncert = StdDevUncertainty(np.sqrt(ccd.data), unit='m') ccd._uncertainty = uncert ccd._uncertainty._parent_nddata = ccd with pytest.raises(ValueError): _extract_ccddata(ccd)
def test_simple_import(self): ccd = self.ccd f = _extract_ccddata(ccd) assert_equal(f['data'], 100 * np.ones(self.shape)) assert_equal(f['mask'], self.mask) assert_equal(f['unit'], self.unit) assert_equal(f['meta'], self.meta) assert_equal(f['uncertainty'], 10 * np.ones(self.shape))
def test_inverse_variance_uncert(self): ccd = self.ccd uncert = InverseVariance(0.01 * np.ones(self.shape), unit=self.unit + '-2') ccd.uncertainty = uncert f = _extract_ccddata(ccd) assert_equal(f['data'], 100 * np.ones(self.shape)) assert_equal(f['mask'], self.mask) assert_equal(f['unit'], self.unit) assert_equal(f['meta'], self.meta) assert_equal(f['uncertainty'], 10 * np.ones(self.shape))
def test_invalid_uncert(self): ccd = self.ccd ccd.uncertainty = UnknownUncertainty(np.ones(self.shape)) with pytest.raises(TypeError): _extract_ccddata(ccd)
def test_no_uncertainty(self): ccd = self.ccd ccd.uncertainty = None f = _extract_ccddata(ccd) assert_is_none(f['uncertainty'])
def test_no_mask(self): ccd = self.ccd ccd.mask = None f = _extract_ccddata(ccd) assert_is_none(f['mask'])