Coverage for tests/testMaps.py : 22%

Hot-keys on this page
r m x p toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
"""Generate a simple array of numbers, evenly arranged between min/max, but (optional) random order.""" datavalues = np.arange(0, size, dtype='float') datavalues *= (float(max) - float(min)) / (datavalues.max() - datavalues.min()) datavalues += min if random > 0: rng = np.random.RandomState(random) randorder = rng.rand(size) randind = np.argsort(randorder) datavalues = datavalues[randind] ids = np.arange(size) datavalues = np.array(list(zip(datavalues, datavalues, ids)), dtype=[('fieldRA', 'float'), ('fieldDec', 'float'), ('fieldId', 'int')]) return datavalues
rng = np.random.RandomState(seed) names = ['fieldId', 'fieldRA', 'fieldDec'] types = [int, float, float] fieldData = np.zeros(100, dtype=list(zip(names, types))) fieldData['fieldId'] = np.arange(100) fieldData['fieldRA'] = rng.rand(100) fieldData['fieldDec'] = rng.rand(100) return fieldData
mapPath = os.environ['SIMS_MAPS_DIR']
if os.path.isfile(os.path.join(mapPath, 'DustMaps/dust_nside_128.npz')):
data = makeDataValues(random=981) dustmap = maps.DustMap()
slicer1 = slicers.HealpixSlicer(latLonDeg=False) slicer1.setupSlicer(data) result1 = dustmap.run(slicer1.slicePoints) assert('ebv' in list(result1.keys()))
fieldData = makeFieldData(2234)
slicer2 = slicers.OpsimFieldSlicer(latLonDeg=False) slicer2.setupSlicer(data, fieldData) result2 = dustmap.run(slicer2.slicePoints) assert('ebv' in list(result2.keys()))
# Check interpolation works dustmap = maps.DustMap(interp=True) result3 = dustmap.run(slicer2.slicePoints) assert('ebv' in list(result3.keys()))
# Check warning gets raised dustmap = maps.DustMap(nside=4) with warnings.catch_warnings(record=True) as w: warnings.simplefilter("always") dustmap.run(slicer1.slicePoints) self.assertIn("nside", str(w[-1].message)) else: warnings.warn('Did not find dustmaps, not running testMaps.py')
mapPath = os.environ['SIMS_MAPS_DIR']
if os.path.isfile(os.path.join(mapPath, 'StarMaps/starDensity_r_nside_64.npz')): data = makeDataValues(random=887) # check that it works if nside does not match map nside of 64 nsides = [32, 64, 128] for nside in nsides: starmap = maps.StellarDensityMap() slicer1 = slicers.HealpixSlicer(nside=nside, latLonDeg=False) slicer1.setupSlicer(data) result1 = starmap.run(slicer1.slicePoints) assert('starMapBins' in list(result1.keys())) assert('starLumFunc' in list(result1.keys())) assert(np.max(result1['starLumFunc'] > 0))
fieldData = makeFieldData(22)
slicer2 = slicers.OpsimFieldSlicer(latLonDeg=False) slicer2.setupSlicer(data, fieldData) result2 = starmap.run(slicer2.slicePoints) assert('starMapBins' in list(result2.keys())) assert('starLumFunc' in list(result2.keys())) assert(np.max(result2['starLumFunc'] > 0))
else: warnings.warn('Did not find stellar density map, skipping test.')
lsst.utils.tests.init()
lsst.utils.tests.init() unittest.main() |