peterf@2: #!/usr/bin/env python
peterf@2: # CREATED:2014-01-18 14:09:05 by Brian McFee <brm2132@columbia.edu>
peterf@2: # unit tests for util routines 
peterf@2: 
peterf@2: import numpy as np
peterf@2: import librosa
peterf@2: 
peterf@2: def test_frame():
peterf@2: 
peterf@2:     # Generate a random time series
peterf@2:     def __test(P):
peterf@2:         frame, hop = P
peterf@2: 
peterf@2:         y = np.random.randn(8000)
peterf@2:         y_frame = librosa.util.frame(y, frame_length=frame, hop_length=hop)
peterf@2: 
peterf@2:         for i in xrange(y_frame.shape[1]):
peterf@2:             assert np.allclose(y_frame[:, i], y[ i * hop : (i * hop + frame)])
peterf@2: 
peterf@2:     for frame in [256, 1024, 2048]:
peterf@2:         for hop_length in [64, 256, 512]:
peterf@2:             yield (__test, [frame, hop_length])