added new tests for ComposeMany

Author: lilywilde

tests/test_compose.py

@@ -1,5 +1,11 @@
 import pytest
-from torchaudio_augmentations import Compose, ComposeMany, RandomResizedCrop
+import torch
+from torchaudio_augmentations import (
+    Compose,
+    ComposeMany,
+    RandomResizedCrop,
+    Reverb,
+)
 
 from .utils import generate_waveform
 
@@ -10,7 +16,11 @@
 @pytest.mark.parametrize("num_channels", [1, 2])
 def test_compose(num_channels):
     audio = generate_waveform(sample_rate, num_samples, num_channels)
-    transform = Compose([RandomResizedCrop(num_samples),])
+    transform = Compose(
+        [
+            RandomResizedCrop(num_samples),
+        ]
+    )
 
     t_audio = transform(audio)
     assert t_audio.shape[0] == num_channels
@@ -19,14 +29,21 @@ def test_compose(num_channels):
 
 @pytest.mark.parametrize("num_channels", [1, 2])
 def test_compose_many(num_channels):
-    num_augmented_samples = 10
+    num_augmented_samples = 4
 
     audio = generate_waveform(sample_rate, num_samples, num_channels)
     transform = ComposeMany(
-        [RandomResizedCrop(num_samples),], num_augmented_samples=num_augmented_samples,
+        [
+            RandomResizedCrop(num_samples),
+            Reverb(sample_rate),
+        ],
+        num_augmented_samples=num_augmented_samples,
     )
 
     t_audio = transform(audio)
     assert t_audio.shape[0] == num_augmented_samples
     assert t_audio.shape[1] == num_channels
     assert t_audio.shape[2] == num_samples
+
+    for n in range(1, num_augmented_samples):
+        assert torch.all(t_audio[0].eq(t_audio[n])) == False