speedup tests

tests/test_augmentations.py

@@ -4,12 +4,12 @@
 from openunmix import data
 
 
-@pytest.fixture(params=[4096, 4096 * 10])
+@pytest.fixture(params=[4096])
 def nb_timesteps(request):
     return int(request.param)
 
 
-@pytest.fixture(params=[1, 2, 3])
+@pytest.fixture(params=[1, 2])
 def nb_channels(request):
     return request.param
 

tests/test_model.py

@@ -8,22 +8,22 @@
 from openunmix import umxl
 
 
-@pytest.fixture(params=[10, 100])
+@pytest.fixture(params=[100])
 def nb_frames(request):
     return int(request.param)
 
 
-@pytest.fixture(params=[1, 2, 3])
+@pytest.fixture(params=[1, 2])
 def nb_channels(request):
     return request.param
 
 
-@pytest.fixture(params=[1, 5])
+@pytest.fixture(params=[2])
 def nb_samples(request):
     return request.param
 
 
-@pytest.fixture(params=[111, 1024])
+@pytest.fixture(params=[1024])
 def nb_bins(request):
     return request.param
 
@@ -33,7 +33,7 @@ def spectrogram(request, nb_samples, nb_channels, nb_bins, nb_frames):
     return torch.rand((nb_samples, nb_channels, nb_bins, nb_frames))
 
 
-@pytest.fixture(params=[True, False])
+@pytest.fixture(params=[False])
 def unidirectional(request):
     return request.param
 

tests/test_transforms.py

@@ -4,27 +4,27 @@
 from openunmix import transforms
 
 
-@pytest.fixture(params=[4096, 44100])
+@pytest.fixture(params=[4096])
 def nb_timesteps(request):
     return int(request.param)
 
 
-@pytest.fixture(params=[1, 2])
+@pytest.fixture(params=[2])
 def nb_channels(request):
     return request.param
 
 
-@pytest.fixture(params=[1, 2])
+@pytest.fixture(params=[2])
 def nb_samples(request):
     return request.param
 
 
-@pytest.fixture(params=[1024, 2048, 4096])
+@pytest.fixture(params=[2048])
 def nfft(request):
     return int(request.param)
 
 
-@pytest.fixture(params=[2, 4])
+@pytest.fixture(params=[2])
 def hop(request, nfft):
     return nfft // request.param