def test_allele_expectation(): p = RandomState(0).rand(2, 3) p = p / p.sum(1)[:, None] assert_allclose( allele_expectation(p, 2, 2), [ [0.9710998244964055, 1.028900175503594], [0.9374325310073951, 1.0625674689926048], ], )
def test_no_coefficients(self): """test that white noise and filtered white noise is equal when no coefficients are specified""" wn_signal = RandomState(seed=1).standard_normal(400) signal = self.fwn.signal() self.assertEqual(wn_signal.sum(), signal.sum())