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
