def test_amplification_FSPL_for_Lyrae():
tau = np.array([0.001])
uo = np.array([0] * len(tau))
rho = 0.01
gamma = 0.5
yoo_table = microlmodels.yoo_table
magnification = microlmagnification.amplification_FSPL(tau, uo, rho, gamma, yoo_table)
assert np.allclose(magnification, np.array([216.97028636]))
def test_amplification_FSPL_two_points():
tau = np.array([0.001, 0.002])
uo = np.array([0] * len(tau))
rho = 0.01
gamma = 0.5
yoo_table = microlmodels.yoo_table
magnification = microlmagnification.amplification_FSPL(tau, uo, rho, gamma, yoo_table)
assert np.allclose(magnification, np.array([216.97028636, 214.44622417]))
def test_amplification_FSPL_two_points():
tau = np.array([0.001, 0.002])
uo = np.array([0] * len(tau))
rho = 0.01
gamma = 0.5
yoo_table = microlmodels.yoo_table
magnification, u = microlmagnification.amplification_FSPL(tau, uo, rho, gamma, yoo_table)
assert np.allclose(magnification, np.array([216.97028636, 214.44622417]))
assert np.allclose(u, np.array([0.001, 0.002]))