def test_cdf():
jax_dist = Logistic(loc=10, scale=1)
original_dist = jax_dist.rv()
for x in range(5, 15):
assert jax_dist.cdf(x) == pytest.approx(original_dist.cdf(x), rel=0.1)
def test_fit_single_scipy():
dist = Logistic.from_samples_scipy(onp.array([0.1, 0.2]))
assert dist.loc == pytest.approx(0.15, abs=0.02)
def test_fit_single_compare():
scipy_dist = Logistic.from_samples_scipy(onp.array([0.1, 0.2]))
jax_dist = Logistic.from_samples(np.array([0.1, 0.2]))
assert scipy_dist.loc == pytest.approx(float(jax_dist.loc), abs=0.1)
assert scipy_dist.scale == pytest.approx(float(jax_dist.scale), abs=0.1)