def test_longDurations():
"""Test that bls is not computed if transit duration is too long"""
time,flux = loadData()
periodList = [10.]
duration_daysList = [11]
blsArray = fbls.computeBlsForManyPeriods(time, flux, duration_daysList, periodList)
assert(blsArray.shape[:2] == (1,1))
assert(np.all(blsArray==0))
def test_fbls1(plot=False):
time, flux = makeTestData()
duration_daysList = np.array([2,4,6,8]).astype(float) / 24.
periodList = [2., 2.3, 2.4, 2.41, 2.42, 2.5, 3.0]
blsArray = fbls.computeBlsForManyPeriods(time, flux, duration_daysList, periodList)
index = np.argmax(blsArray)
per, epc, dur= fbls.getParamsOfIndex(blsArray, index,\
duration_daysList, periodList)
if plot:
print per, epc+time[0], dur*24
makePlots(time, flux, periodList, blsArray, per, epc)
assert(per> 2.3 and per < 2.42)
assert(int(dur*24) == 6)
assert(epc >1.09 and epc < 1.11)
return blsArray
