def test_rate_to_probability_symmetry():
rate = 0.0001
for _ in range(100):
prob = rate_to_probability(rate)
assert round(rate, 5) == round(probability_to_rate(prob), 5)
rate += (1-0.0001)/100.0
def test_rate_to_probability_vectorizability():
rate = 0.001
rate = np.array([rate]*100)
prob = rate_to_probability(rate)
assert round(prob[10], 5) == round(0.00099950016662497809, 5)
assert round(np.sum(rate), 5) == round(np.sum(probability_to_rate(prob)), 5)
def test_probability_to_rate():
prob = 0.00099950016662497809
rate = probability_to_rate(prob)
assert round(rate, 5) == round(0.001, 5)