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)