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
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