def test_sine(): for period in PERIODS: s = sequences.sine(1.0, period, CYCLES, period / 3) p = periodogram_peaks(s, thresh=1.0) assert len(p) == 1 and p[0][2] <= period and p[0][3] >= period
def test_square_wave(): for period in PERIODS: s = sequences.impulses(period / 5, period, CYCLES, period / 3) p = periodogram_peaks(s, thresh=1.0) assert len(p) == 1 and p[0][2] <= period and p[0][3] >= period
def test_zeros(): assert periodogram_peaks(np.zeros(1000)) == None
def test_sine(): for period in PERIODS: s = sequences.sine(1.0, period, CYCLES, period // 3) p = periodogram_peaks(s, thresh=1.0) assert len(p) == 1 and p[0][2] <= period and p[0][3] >= period
def test_square_wave(): for period in PERIODS: s = sequences.impulses(period // 5, period, CYCLES, period // 3) p = periodogram_peaks(s, thresh=1.0) assert len(p) == 1 and p[0][2] <= period and p[0][3] >= period