def test_isWhite(self): notwhite = Correlator([.1] * 100) assert not notwhite.isWhite('mehra') assert not notwhite.isWhite('ljung-box') white = Correlator([0] * 50 + [1] + [0] * 50) assert white.isWhite('mehra') assert white.isWhite('ljung-box') with pytest.raises(ValueError): white.isWhite('novalidmethod')
from numpy.random import randn from noiseestimation.correlator import Correlator num_runs = 50 arr_length = 200 num_mehra_white = 0 num_ljungbox_white = 0 for i in range(num_runs): seq = randn(arr_length) cor = Correlator(seq) if cor.isWhite(): num_ljungbox_white += 1 if cor.isWhite('mehra'): num_mehra_white += 1 print("-" * 10) print("%d / %d white (%.3f %%) using Mehra method" % (num_mehra_white, num_runs, 100 * float(num_mehra_white) / num_runs)) print( "%d / %d white (%.3f %%) using Ljund-Box method" % (num_ljungbox_white, num_runs, 100 * float(num_ljungbox_white) / num_runs))