Chris@8: 
Chris@8: import tempo_estimator as est
Chris@11: import numpy as np
Chris@11: 
Chris@11: def test_detection_function():
Chris@11:     samples = np.array([1,0, 0,0, 0,0, 1,0, 0,0])
Chris@11:     df = est.detection_function(samples, 2)
Chris@11:     assert len(df) == 5
Chris@11:     assert df[3] > df[2]
Chris@11:     assert df[0] > df[2]
Chris@8: 
Chris@8: def test_tempo_120bpm():
Chris@8:     tempo = est.estimate_tempo_of_file('testfiles/120bpm.wav')
Chris@8:     assert abs(tempo - 120.0) < 0.5
Chris@8: 
Chris@8: