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