def test_weighted_sample_unnormalized(self): disc = Discrete(4) weights = np.array([1., 2., 3., 5.]) res = disc.weighted_sample(weights) assert res >= 0 and res < disc.n
def test_weighted_sample(self): disc = Discrete(4) weights = [0.1, 0.2, 0.3, 0.4] res = disc.weighted_sample(weights) assert res >= 0 and res < disc.n