def test_weighted_4(): ex = ExpectationWeighted() ex.update(np.array([1, 1]), np.array([0, 101])) ex.update(np.array([2, 2]), np.array([23, 0])) assert_allclose([2, 1], ex.get_value()) assert_allclose([23, 101], ex.get_mass())
def test_weighted_1(): ex = ExpectationWeighted() ex.update(np.array([1, 2]), np.array([0, 0])) ex.update(np.array([1, 2]), np.array([0, 0])) assert np.all(np.isnan(ex.get_value()))
def test_weighted_2(): ex = ExpectationWeighted() ex.update(np.array([1, 1]), np.array([0, 0])) ex.update(np.array([1, 2]), np.array([0, 0])) assert_allclose(42, ex.get_value(42))