def test_log1mexp_deprecation_warnings(): with pytest.warns( FutureWarning, match="pymc.math.log1mexp_numpy will expect a negative input", ): res_pos = log1mexp_numpy(2) with pytest.warns(None) as record: res_neg = log1mexp_numpy(-2, negative_input=True) assert not record with pytest.warns( FutureWarning, match="pymc.math.log1mexp will expect a negative input", ): res_pos_at = log1mexp(2).eval() with pytest.warns(None): res_neg_at = log1mexp(-2, negative_input=True).eval() assert np.isclose(res_pos, res_neg) assert np.isclose(res_pos_at, res_neg) assert np.isclose(res_neg_at, res_neg)
def test_log1mexp(): vals = np.array([-1, 0, 1e-20, 1e-4, 10, 100, 1e20]) vals_ = vals.copy() # import mpmath # mpmath.mp.dps = 1000 # [float(mpmath.log(1 - mpmath.exp(-x))) for x in vals] expected = np.array([ np.nan, -np.inf, -46.051701859880914, -9.210390371559516, -4.540096037048921e-05, -3.720075976020836e-44, 0.0, ]) actual = at.log1mexp(-vals).eval() npt.assert_allclose(actual, expected) actual_ = log1mexp_numpy(-vals, negative_input=True) npt.assert_allclose(actual_, expected) # Check that input was not changed in place npt.assert_allclose(vals, vals_)
def test_log1mexp_numpy_integer_input(): assert np.isclose(log1mexp_numpy(-2, negative_input=True), at.log1mexp(-2).eval())
def test_log1mexp_numpy_no_warning(): """Assert RuntimeWarning is not raised for very small numbers""" with pytest.warns(None) as record: log1mexp_numpy(-1e-25, negative_input=True) assert not record