fix test code for 0.3.3

Author: seyong92

tests/test_pv.py

@@ -0,0 +1,75 @@
+import pytest
+import pytsmod as tsm
+from pytsmod.pvtsm import _find_peaks
+import numpy as np
+from scipy.io import loadmat
+import soundfile as sf
+
+
+@pytest.mark.parametrize('alpha', [0.75, 1, 1.25])
+@pytest.mark.parametrize('win_type', ['hann', 'sin'])
+@pytest.mark.parametrize('win_size', [2048, 4096])
+@pytest.mark.parametrize('syn_hop_size', [512, 1024])
+@pytest.mark.parametrize('phase_lock', [1, 0])
+def test_pv_fixed(alpha, win_type, win_size, syn_hop_size, phase_lock):
+    x, _ = sf.read('tests/data/castanetsviolin.wav')
+    matlab_results = loadmat('tests/data/pv_fixed_rate.mat')['result']
+
+    y = tsm.phase_vocoder(x, alpha, win_type, win_size, syn_hop_size,
+                          phase_lock=phase_lock)
+
+    _, w = np.where(matlab_results[0, :] == np.array([[alpha]]))
+    matlab_results = matlab_results[:, w]
+    _, w = np.where(matlab_results[1, :] == np.array([[2 if win_type == 'hann' else 1]]))
+    matlab_results = matlab_results[:, w]
+    _, w = np.where(matlab_results[2, :] == np.array([[win_size]]))
+    matlab_results = matlab_results[:, w]
+    _, w = np.where(matlab_results[3, :] == np.array([[syn_hop_size]]))
+    matlab_results = matlab_results[:, w]
+    _, w = np.where(matlab_results[4, :] == np.array([[phase_lock]]))
+    matlab_results = matlab_results[:, w]
+
+    y_matlab = matlab_results[5, :][0].squeeze()
+    print(np.max(y - y_matlab))
+    assert np.allclose(y, y_matlab)
+
+
+# @pytest.mark.parametrize('win_type', ['hann', 'sin'])
+# @pytest.mark.parametrize('win_size', [1024, 2048])
+# @pytest.mark.parametrize('syn_hop_size', [512, 1024])
+# @pytest.mark.parametrize('tolerance', [512, 1024])
+# def test_pv_nonlinear(win_type, win_size, syn_hop_size, tolerance):
+#     x, _ = sf.read('tests/data/beethovenorchestra.wav')
+#     ap = loadmat('tests/data/anchorpoints.mat')['anchorpoints'].T - 1
+#     wsola_nonlinear = loadmat('tests/data/wsola_nonlinear.mat')
+
+#     y = tsm.wsola(x, ap, win_type, win_size, syn_hop_size, tolerance)
+
+#     sample_name = f'orc_{win_type}_{win_size}_{syn_hop_size}_{tolerance}'
+#     y_matlab = wsola_nonlinear[sample_name].squeeze()
+#     assert np.allclose(y, y_matlab)
+
+
+# @pytest.mark.parametrize('n_chan', range(2, 9))
+# @pytest.mark.parametrize('transpose', [True, False])
+# def test_pv_multichannel(n_chan, transpose):
+#     x, _ = sf.read('tests/data/dogbeetle.wav')
+#     x_wsola = tsm.wsola(x, 1.3)
+
+#     x_multi = np.tile(x, (n_chan, 1))
+
+#     if transpose:
+#         x_multi = x_multi.T
+
+#     x_multi_wsola = tsm.wsola(x_multi, 1.3)
+
+#     for i in range(x_multi_wsola.shape[0]):
+#         assert np.allclose(x_wsola, x_multi_wsola[i, :])
+
+
+def test_find_peaks():
+    sample = np.array([0, 0.3, 1, 0.4, 0.3, 2, 0.3, 3, 0.1, 0.2, 0.14, 2])
+    peaks, infl = _find_peaks(sample)
+
+    assert np.allclose(peaks, np.array([2, 7, 11]))
+    assert np.allclose(infl, np.array([[0, 5, 9], [4, 8, 11]]))

tests/test_pytsmod.py

@@ -2,4 +2,5 @@
 
 
 def test_version():
-    assert __version__ == '0.3.2'
+    assert __version__ == '0.3.3'
+