def _select_targets_file(filename):
'''Returns targets in filename that pass the cuts'''
from desitarget import io
objects = io.read_tractor(filename, columns=columns)
keep = apply_cuts(objects)
return io.fix_tractor_dr1_dtype(objects[keep])
def _select_targets_file(filename):
'''Returns targets in filename that pass the cuts'''
from desitarget import io
objects = io.read_tractor(filename, columns=columns)
keep = apply_cuts(objects)
return io.fix_tractor_dr1_dtype(objects[keep])
def test_fix_dr1(self):
'''test the DR1 TYPE dype fix (make everything S4)'''
# - First, break it
files = io.list_sweepfiles(self.datadir)
objects = io.read_tractor(files[0])
dt = objects.dtype.descr
for i in range(len(dt)):
if dt[i][0] == 'TYPE':
dt[i] = ('TYPE', 'S10')
break
badobjects = objects.astype(np.dtype(dt))
newobjects = io.fix_tractor_dr1_dtype(badobjects)
self.assertEqual(newobjects['TYPE'].dtype, np.dtype('S4'))
def _finalize_targets(objects, desi_target, bgs_target, mws_target):
#- desi_target includes BGS_ANY and MWS_ANY, so we can filter just
#- on desi_target != 0
keep = (desi_target != 0)
objects = objects[keep]
desi_target = desi_target[keep]
bgs_target = bgs_target[keep]
mws_target = mws_target[keep]
#- Add *_target mask columns
targets = desitarget.targets.finalize(objects, desi_target, bgs_target,
mws_target)
return io.fix_tractor_dr1_dtype(targets)
def test_fix_dr1(self):
'''test the DR1 TYPE dype fix (make everything S4)'''
#- First, break it
files = io.list_sweepfiles(self.datadir)
objects = io.read_tractor(files[0])
dt = objects.dtype.descr
for i in range(len(dt)):
if dt[i][0] == 'TYPE':
dt[i] = ('TYPE', 'S10')
break
badobjects = objects.astype(np.dtype(dt))
newobjects = io.fix_tractor_dr1_dtype(badobjects)
self.assertEqual(newobjects['TYPE'].dtype, np.dtype('S4'))
def _select_targets_file(filename):
'''Returns targets in filename that pass the cuts'''
from desitarget import io
objects = io.read_tractor(filename)
desi_target, bgs_target, mws_target = apply_cuts(objects)
#- desi_target includes BGS_ANY and MWS_ANY, so we can filter just
#- on desi_target != 0
keep = (desi_target != 0)
objects = objects[keep]
desi_target = desi_target[keep]
bgs_target = bgs_target[keep]
mws_target = mws_target[keep]
#- Add *_target mask columns
targets = desitarget.targets.finalize(
objects, desi_target, bgs_target, mws_target)
return io.fix_tractor_dr1_dtype(targets)