def test_BinnedNLL_mask(): c = BinnedNLL([5, 1000, 1], [0, 1, 2, 3], expon_cdf) c_unmasked = c(1) c.mask = np.arange(3) != 1 assert c(1) < c_unmasked
def test_BinnedNLL_mask(): c = BinnedNLL([1, 1000, 2], [0, 1, 2, 3], lambda x, a: x + a) assert c(2) == pytest.approx(-7000, rel=0.1) c.mask = np.arange(3) != 1 assert c(2) == pytest.approx(-3, rel=0.1)