From c4a17027be9060e97b7ab056a9fb00679038bf64 Mon Sep 17 00:00:00 2001
From: jachymuv <17248068+jachymuv@users.noreply.github.com>
Date: Sat, 22 May 2021 05:19:44 +0100
Subject: [PATCH] Do not use IO functions in batch consistency test (#1521)

---
 .../transforms/batch_consistency_test.py       | 18 ++++++------------
 1 file changed, 6 insertions(+), 12 deletions(-)

diff --git a/test/torchaudio_unittest/transforms/batch_consistency_test.py b/test/torchaudio_unittest/transforms/batch_consistency_test.py
index cab68e0152..609f1ee631 100644
--- a/test/torchaudio_unittest/transforms/batch_consistency_test.py
+++ b/test/torchaudio_unittest/transforms/batch_consistency_test.py
@@ -74,8 +74,7 @@ def test_batch_compute_deltas(self):
         self.assertEqual(computed, expected)
 
     def test_batch_mulaw(self):
-        test_filepath = common_utils.get_asset_path('steam-train-whistle-daniel_simon.wav')
-        waveform, _ = torchaudio.load(test_filepath)  # (2, 278756), 44100
+        waveform = common_utils.get_whitenoise(sample_rate=8000, duration=1, n_channels=2)
 
         # Single then transform then batch
         waveform_encoded = torchaudio.transforms.MuLawEncoding()(waveform)
@@ -99,8 +98,7 @@ def test_batch_mulaw(self):
         self.assertEqual(computed, expected)
 
     def test_batch_spectrogram(self):
-        test_filepath = common_utils.get_asset_path('steam-train-whistle-daniel_simon.wav')
-        waveform, _ = torchaudio.load(test_filepath)  # (2, 278756), 44100
+        waveform = common_utils.get_whitenoise(sample_rate=8000, duration=1, n_channels=2)
 
         # Single then transform then batch
         expected = torchaudio.transforms.Spectrogram()(waveform).repeat(3, 1, 1, 1)
@@ -110,8 +108,7 @@ def test_batch_spectrogram(self):
         self.assertEqual(computed, expected)
 
     def test_batch_melspectrogram(self):
-        test_filepath = common_utils.get_asset_path('steam-train-whistle-daniel_simon.wav')
-        waveform, _ = torchaudio.load(test_filepath)  # (2, 278756), 44100
+        waveform = common_utils.get_whitenoise(sample_rate=8000, duration=1, n_channels=2)
 
         # Single then transform then batch
         expected = torchaudio.transforms.MelSpectrogram()(waveform).repeat(3, 1, 1, 1)
@@ -121,8 +118,7 @@ def test_batch_melspectrogram(self):
         self.assertEqual(computed, expected)
 
     def test_batch_mfcc(self):
-        test_filepath = common_utils.get_asset_path('steam-train-whistle-daniel_simon.wav')
-        waveform, _ = torchaudio.load(test_filepath)
+        waveform = common_utils.get_whitenoise(sample_rate=8000, duration=1, n_channels=2)
 
         # Single then transform then batch
         expected = torchaudio.transforms.MFCC()(waveform).repeat(3, 1, 1, 1)
@@ -160,8 +156,7 @@ def test_batch_TimeStretch(self, test_pseudo_complex):
         self.assertEqual(computed, expected, atol=1e-5, rtol=1e-5)
 
     def test_batch_Fade(self):
-        test_filepath = common_utils.get_asset_path('steam-train-whistle-daniel_simon.wav')
-        waveform, _ = torchaudio.load(test_filepath)  # (2, 278756), 44100
+        waveform = common_utils.get_whitenoise(sample_rate=8000, duration=1, n_channels=2)
         fade_in_len = 3000
         fade_out_len = 3000
 
@@ -173,8 +168,7 @@ def test_batch_Fade(self):
         self.assertEqual(computed, expected)
 
     def test_batch_Vol(self):
-        test_filepath = common_utils.get_asset_path('steam-train-whistle-daniel_simon.wav')
-        waveform, _ = torchaudio.load(test_filepath)  # (2, 278756), 44100
+        waveform = common_utils.get_whitenoise(sample_rate=8000, duration=1, n_channels=2)
 
         # Single then transform then batch
         expected = torchaudio.transforms.Vol(gain=1.1)(waveform).repeat(3, 1, 1)