Add unit test for librosa stft

- Add parity check with torch.stft and librosa.stft
BriansIDP · Dec 18, 2022 · 7931f2e · 7931f2e
1 parent 743821d
commit 7931f2e
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/test/espnet2/layers/test_stft.py b/test/espnet2/layers/test_stft.py
@@ -38,3 +38,18 @@ def test_inverse():
     x_lengths = torch.IntTensor([400, 300])
     raw, _ = layer.inverse(y, x_lengths)
     raw, _ = layer.inverse(y)
+
+
+def test_librosa_stft():
+    mkl_is_available = torch.backends.mkl.is_available()
+    if not mkl_is_available:
+        raise RuntimeError("MKL is not available.")
+
+    layer = Stft()
+    layer.eval()
+    x = torch.randn(2, 16000, device="cpu")
+    y_torch, _ = layer(x)
+    torch._C.has_mkl = False
+    y_librosa, _ = layer(x)
+    assert torch.allclose(y_torch, y_librosa, atol=7e-6)
+    torch._C.has_mkl = True