def test_uniform_scalar_distribution(): pmf = np.array([1/3] * 3) outcomes = (0, 1, 2) dit.math.prng.seed(1) d = dit.uniform_scalar_distribution(len(outcomes)) assert_equal(d.outcomes, outcomes) np.testing.assert_allclose(d.pmf, pmf) dit.math.prng.seed(1) d = dit.uniform_scalar_distribution(outcomes) assert_equal(d.outcomes, outcomes) np.testing.assert_allclose(d.pmf, pmf)
def test_uniform_scalar_distribution(): pmf = np.array([1 / 3] * 3) outcomes = (0, 1, 2) dit.math.prng.seed(1) d = dit.uniform_scalar_distribution(len(outcomes)) assert_equal(d.outcomes, outcomes) np.testing.assert_allclose(d.pmf, pmf) dit.math.prng.seed(1) d = dit.uniform_scalar_distribution(outcomes) assert_equal(d.outcomes, outcomes) np.testing.assert_allclose(d.pmf, pmf)