peterf@2: #!/usr/bin/env python peterf@2: # CREATED: 2013-10-06 22:31:29 by Dawen Liang peterf@2: # unit tests for librosa.decompose peterf@2: peterf@2: import numpy as np peterf@2: import librosa peterf@2: peterf@2: peterf@2: def test_default_decompose(): peterf@2: X = np.array([[1, 2, 3, 4, 5, 6], [1, 1, 1.2, 1, 0.8, 1]]) peterf@2: (W, H) = librosa.decompose.decompose(X) peterf@2: assert np.allclose(X, W.dot(H), rtol=1e-2, atol=1e-2) peterf@2: pass