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