#### Real code starts here print "\n\n\n ***Warning*** --- this program takes a few minutes to run --- \n\n" # filter set of choice is the LSST filters listOfFilters = 'LSST.filters' pathToFilters = '../filter_data' # create a dictionary: keyword is filter name, value is a Filter object filterDict = sedFilter.createFilterDict(listOfFilters, pathToFilters) # return the filter names filterList = sedFilter.getFilterList(listOfFilters, pathToFilters) print 'Filter list = ', filterList nFilter = len(filterList) # load in elliptical SED (1st col: wavelength in A, 2nd col: fluxes in wl units) seddata = np.loadtxt('../sed_data/El_B2004a.sed') # turn into SED object sed = sedFilter.SED(seddata[:,0], seddata[:,1]) # instantiate photometry calculations # need cosmological model
import numpy as np import photometry as phot import sedFilter import cosmo import photErrorModel as pem # Single SED for testing fname = os.path.join(os.path.dirname(__file__), '../sed_data/El_B2004a.sed') seddata = np.loadtxt(fname) TEST_SED = sedFilter.SED(seddata[:,0], seddata[:,1]) # List of filters for testing listOfFilters = 'LSST.filters' pathToFilters = '../filter_data' TEST_FILTERDICT = sedFilter.createFilterDict(listOfFilters, pathToFilters) FILTERLIST = sedFilter.getFilterList(listOfFilters, pathToFilters) # List of SEDs for testing SEDLIST = [] sedfiles = ['../sed_data/El_B2004a.sed','../sed_data/Sbc_B2004a.sed','../sed_data/Scd_B2004a.sed', '../sed_data/Im_B2004a.sed'] for name in sedfiles: seddata = np.loadtxt(name) SEDLIST.append(sedFilter.SED(seddata[:,0], seddata[:,1])) # Cosmological model COSMOMODEL = cosmo.cosmologyCalculator() class TestPhotCalcs(unittest.TestCase):