Exemple #1
0
def test_calculate_procreation_rates_2():
    pf = ParticleFilter(func, 100)

    x = np.random.rand(10, 2)
    y = np.ones(10)
    pf.set_seed(x, y)

    rates = pf._calculate_procreation_rates(y, 10000)
    assert len(np.unique(rates)) == 1
    assert rates[0] > 0

    rates = pf._calculate_procreation_rates(y, 13)
    assert len(np.unique(rates)) == 2
    assert np.amax(rates) - np.amin(rates) == 1
    assert np.sum(rates) == 13
Exemple #2
0
def test_calculate_procreation_rates_1():
    pf = ParticleFilter(func, 100)

    x = np.random.rand(10, 2)
    y = np.arange(10)
    pf.set_seed(x, y)

    rates = pf._calculate_procreation_rates(y, 10000)
    rates[:-1] = rates[:-1] - rates[1:]
    positive = np.sum(rates[:-1] > 0)
    assert positive == len(y) - 1