From fc33bd83b68757cff1a3c73501cb3c7bb34740e8 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Mon, 7 Nov 2022 10:25:54 +0530
Subject: [PATCH] transforms test

---
 tests/transforms_test.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/transforms_test.py b/tests/transforms_test.py
index 3053b09..89425ad 100644
--- a/tests/transforms_test.py
+++ b/tests/transforms_test.py
@@ -12,3 +12,7 @@ def test_stft_istft():
         spectrogram = stft(sample_input)
         waveform = istft(spectrogram)
     assert sample_input.shape == waveform.shape
+    assert (
+        torch.isclose(waveform, sample_input).sum().item()
+        > sample_input.shape[-1] // 2
+    )