class TestBitMask(unittest.TestCase):
"""Test desiutil.bitmask.
"""
def setUp(self):
self.ccdmask = BitMask('ccdmask', _bitdefs)
def tearDown(self):
pass
def test_names(self):
"""Test consistency for names to bits to masks.
"""
m = self.ccdmask
for name in m.names():
self.assertEqual(m.mask(name), 2**m.bitnum(name),
'Failed matching mask to bitnum for ' + name)
self.assertEqual(m.mask(name), m.mask(m.bitnum(name)),
'Failed matching mask to name for ' + name)
self.assertEqual(
m.bitname(m.bitnum(name)), name,
'Failed bit name->num->name roundtrip for ' + name)
self.assertEqual(m[name], m[name].mask)
self.assertEqual(m.bitname(name), m[name].name)
self.assertEqual(m.bitnum(name), m[name].bitnum)
c = m.comment(name)
names = m.names(m.COSMIC | m.BAD | 2**13)
self.assertTrue('COSMIC' in names)
self.assertTrue('BAD' in names)
self.assertTrue('UNKNOWN13' in names)
def test_mask(self):
"""Test options for blat.mask().
"""
for i in range(4):
self.assertEqual(self.ccdmask.mask(i), 2**i)
m = self.ccdmask
self.assertEqual(m.mask('BAD|COSMIC'), m.BAD | m.COSMIC)
def test_access(self):
"""Miscellaneous stuff that should work.
"""
self.assertEqual(self.ccdmask._name, 'ccdmask')
self.assertEqual(self.ccdmask['HOT'].blat, 'foo')
self.assertEqual(self.ccdmask.HOT.blat, 'foo')
self.assertEqual(self.ccdmask.HOT.name, 'HOT')
self.assertEqual(self.ccdmask.HOT.bitnum, 1)
self.assertEqual(self.ccdmask.HOT.mask, 2)
self.assertEqual(self.ccdmask.HOT, 2)
self.assertEqual(self.ccdmask.HOT.comment, "Hot pixel")
self.ccdmask.names()
def test_badname(self):
"""Test raising AttributeError for bad names.
"""
with self.assertRaises(AttributeError):
x = self.ccdmask.BLATFOO
# Attribute already in use
with self.assertRaises(AttributeError):
bit = _MaskBit('BAD', 0, "comment", extra={'real': 'foo'})
# has extra that isn't a dict
with self.assertRaises(ValueError):
blat = BitMask('ccdmask', _baddef1)
def test_str(self):
"""Verify yaml-ness of string representation"""
bitmask = BitMask('ccdmask', yaml.load(str(self.ccdmask)))
self.assertEqual(bitmask.names(), self.ccdmask.names())
for name in bitmask.names():
self.assertEqual(bitmask[name].mask, self.ccdmask[name].mask)
self.assertEqual(bitmask[name].comment, self.ccdmask[name].comment)
self.assertEqual(bitmask[name].bitnum, self.ccdmask[name].bitnum)
self.assertEqual(bitmask[name]._extra, self.ccdmask[name]._extra)
def test_print(self):
"""Test string representations.
"""
ccdmask_repr = """ccdmask:
- [BAD , 0, "Pre-determined bad pixel (any reason)"]
- [HOT , 1, "Hot pixel"]
- [DEAD , 2, "Dead pixel"]
- [SATURATED , 3, "Saturated pixel from object"]
- [COSMIC , 4, "Cosmic ray"]"""
bit_str = (
("BAD bit 0 mask 0x1 - Pre-determined bad pixel " +
"(any reason)"), "HOT bit 1 mask 0x2 - Hot pixel",
"DEAD bit 2 mask 0x4 - Dead pixel",
"SATURATED bit 3 mask 0x8 - Saturated pixel from object",
"COSMIC bit 4 mask 0x10 - Cosmic ray")
bit_repr = ('1', '2', '4', '8', '16')
blat = repr(self.ccdmask)
self.assertEqual(blat, _bitdefyaml)
for i, name in enumerate(self.ccdmask.names()):
self.assertEqual(str(self.ccdmask[name]), bit_str[i])
self.assertEqual(repr(self.ccdmask[name]), bit_repr[i])
tile)
_fits = fits.open(fname)
dat = Table(_fits[1].data)['TARGETID', 'BRICK_OBJID', 'RA', 'DEC',
'BRICKID', 'SV1_DESI_TARGET',
'SV1_BGS_TARGET']
except:
print('Unable to retrieve {}.'.format(fname))
continue
isbgs = (dat['SV1_DESI_TARGET'] & 2**desi_mask.bitnum('BGS_ANY')) != 0
exclude = (
dat['SV1_DESI_TARGET']
& desi_mask.mask('BRIGHT_OBJECT|IN_BRIGHT_OBJECT|NEAR_BRIGHT_OBJECT')
!= 0)
print('Tile {} excludes {} for bright mask.'.format(
tile, len(dat[exclude])))
## print(dat.columns)
## print(result)
## Light weight viewer friendly version.
viewer = dat['RA', 'DEC']
## https://github.com/legacysurvey/decals-web/blob/master/templates/index.html#L1080
## https://github.com/legacysurvey/decals-web/blob/master/templates/index.html#L593
viewer['color'] = Column(data=['black'] * len(dat),
dtype='S16',
class TestBitMask(unittest.TestCase):
"""Test desiutil.bitmask.
"""
def setUp(self):
self.ccdmask = BitMask('ccdmask', _bitdefs)
def tearDown(self):
pass
def test_names(self):
"""Test consistency for names to bits to masks.
"""
m = self.ccdmask
for name in m.names():
self.assertEqual(m.mask(name), 2**m.bitnum(name),
'Failed matching mask to bitnum for ' + name)
self.assertEqual(m.mask(name), m.mask(m.bitnum(name)),
'Failed matching mask to name for ' + name)
self.assertEqual(m.bitname(m.bitnum(name)), name,
'Failed bit name->num->name roundtrip for ' +
name)
self.assertEqual(m[name], m[name].mask)
self.assertEqual(m.bitname(name), m[name].name)
self.assertEqual(m.bitnum(name), m[name].bitnum)
c = m.comment(name)
names = m.names(m.COSMIC | m.BAD | 2**13)
self.assertTrue('COSMIC' in names)
self.assertTrue('BAD' in names)
self.assertTrue('UNKNOWN13' in names)
def test_mask(self):
"""Test options for blat.mask().
"""
for i in range(4):
self.assertEqual(self.ccdmask.mask(i), 2**i)
m = self.ccdmask
self.assertEqual(m.mask('BAD|COSMIC'), m.BAD | m.COSMIC)
def test_access(self):
"""Miscellaneous stuff that should work.
"""
self.assertEqual(self.ccdmask._name, 'ccdmask')
self.assertEqual(self.ccdmask['HOT'].blat, 'foo')
self.assertEqual(self.ccdmask.HOT.blat, 'foo')
self.assertEqual(self.ccdmask.HOT.name, 'HOT')
self.assertEqual(self.ccdmask.HOT.bitnum, 1)
self.assertEqual(self.ccdmask.HOT.mask, 2)
self.assertEqual(self.ccdmask.HOT, 2)
self.assertEqual(self.ccdmask.HOT.comment, "Hot pixel")
self.ccdmask.names()
def test_badname(self):
"""Test raising AttributeError for bad names.
"""
with self.assertRaises(AttributeError):
x = self.ccdmask.BLATFOO
# Attribute already in use
with self.assertRaises(AttributeError):
bit = _MaskBit('BAD', 0, "comment", extra={'real': 'foo'})
# has extra that isn't a dict
with self.assertRaises(ValueError):
blat = BitMask('ccdmask', _baddef1)
def test_str(self):
"""Verify yaml-ness of string representation"""
bitmask = BitMask('ccdmask', yaml.load(str(self.ccdmask)))
self.assertEqual(bitmask.names(), self.ccdmask.names())
for name in bitmask.names():
self.assertEqual(bitmask[name].mask, self.ccdmask[name].mask)
self.assertEqual(bitmask[name].comment, self.ccdmask[name].comment)
self.assertEqual(bitmask[name].bitnum, self.ccdmask[name].bitnum)
self.assertEqual(bitmask[name]._extra, self.ccdmask[name]._extra)
def test_print(self):
"""Test string representations.
"""
ccdmask_repr = """ccdmask:
- [BAD , 0, "Pre-determined bad pixel (any reason)"]
- [HOT , 1, "Hot pixel"]
- [DEAD , 2, "Dead pixel"]
- [SATURATED , 3, "Saturated pixel from object"]
- [COSMIC , 4, "Cosmic ray"]"""
bit_str = (
("BAD bit 0 mask 0x1 - Pre-determined bad pixel " +
"(any reason)"),
"HOT bit 1 mask 0x2 - Hot pixel",
"DEAD bit 2 mask 0x4 - Dead pixel",
"SATURATED bit 3 mask 0x8 - Saturated pixel from object",
"COSMIC bit 4 mask 0x10 - Cosmic ray")
bit_repr = ('1', '2', '4', '8', '16')
blat = repr(self.ccdmask)
self.assertEqual(blat, _bitdefyaml)
for i, name in enumerate(self.ccdmask.names()):
self.assertEqual(str(self.ccdmask[name]), bit_str[i])
self.assertEqual(repr(self.ccdmask[name]), bit_repr[i])