bug fixed in test

neuromorphs · Dec 6, 2023 · f57d4d5 · f57d4d5
1 parent 8dc340b
commit f57d4d5
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/test/test_audio_augmentations.py b/test/test_audio_augmentations.py
@@ -68,7 +68,7 @@ def test_random_pitch_shift():
     sl = 1  # sample length
     data = np.random.rand(1, sr * sl)
 
-    aug = RandomPitchShift(sample_length=sl, samplerate=sl)
+    aug = RandomPitchShift(samplerate=sr)
     pitch_shifted = aug(data)
 
     assert pitch_shifted is not data
@@ -92,9 +92,7 @@ def test_random_amplitude_scale():
     data = np.ones((1, sr * sl))
     min_amp, max_amp = 0.05, 0.15
 
-    aug = RandomAmplitudeScale(
-        sample_length=sl, samplerate=sl, min_amp=min_amp, max_amp=max_amp
-    )
+    aug = RandomAmplitudeScale(samplerate=sr, min_amp=min_amp, max_amp=max_amp)
     amp_scaled = aug(data)
 
     assert amp_scaled is not data