Exemple #1
0
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)
Exemple #2
0
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)