def print_td_approximants(): """Only list the time-domain approximants in lalsimulation. """ napprox = lalsimulation.NumApproximants for i in range(napprox): if lalsimulation.SimInspiralImplementedTDApproximants(i): print i, lalsimulation.GetStringFromApproximant(i)
import pytest import lalsimulation as lalsim # Function to get the name of a function def get_name(f): return f.__name__ # Set of functions to test func_list = [ lalsim.SimInspiralGetSpinSupportFromApproximant, lalsim.SimInspiralGetSpinFreqFromApproximant, lalsim.SimInspiralApproximantAcceptTestGRParams ] # Loop over all implemented TD or FD approximants IMPLEMENTED = [ k for k in range(lalsim.NumApproximants) if lalsim.SimInspiralImplementedTDApproximants(k) or lalsim.SimInspiralImplementedFDApproximants(k) ] @pytest.mark.parametrize("func", func_list, ids=get_name) @pytest.mark.parametrize("k", IMPLEMENTED, ids=lalsim.GetStringFromApproximant) def test_wf_property_lists(k, func): func(k) if __name__ == '__main__': args = sys.argv[1:] or ["-v", "-rs", "--junit-xml=junit-wf_property_lists.xml"] sys.exit(pytest.main(args=[__file__] + args))
def _lalsim_sgburst_waveform(**p): hp, hc = lalsimulation.SimBurstSineGaussian(float(p['q']), float(p['frequency']), float(p['hrss']), float(p['eccentricity']), float(p['polarization']), float(p['delta_t'])) hp = TimeSeries(hp.data.data[:], delta_t=hp.deltaT, epoch=hp.epoch) hc = TimeSeries(hc.data.data[:], delta_t=hc.deltaT, epoch=hc.epoch) return hp, hc for approx_enum in xrange(0, lalsimulation.NumApproximants): if lalsimulation.SimInspiralImplementedTDApproximants(approx_enum): approx_name = lalsimulation.GetStringFromApproximant(approx_enum) _lalsim_enum[approx_name] = approx_enum _lalsim_td_approximants[approx_name] = _lalsim_td_waveform for approx_enum in xrange(0, lalsimulation.NumApproximants): if lalsimulation.SimInspiralImplementedFDApproximants(approx_enum): approx_name = lalsimulation.GetStringFromApproximant(approx_enum) _lalsim_enum[approx_name] = approx_enum _lalsim_fd_approximants[approx_name] = _lalsim_fd_waveform # sine-Gaussian burst for approx_enum in xrange(0, lalsimulation.NumApproximants): if lalsimulation.SimInspiralImplementedFDApproximants(approx_enum): approx_name = lalsimulation.GetStringFromApproximant(approx_enum) _lalsim_enum[approx_name] = approx_enum