[test] update pitchshift test

Paul Brossier committed 7 years ago

Commit 07701481 parent d0f19a71 file · +9 −6Browse files

Showing the changes to python (1 of 1 file). Show all

Unified · Split

 #! /usr/bin/env python
 from numpy.testing import TestCase
-from nose2.tools import params
+from _tools import parametrize
 import numpy as np
 import aubio
         with self.assertRaises(RuntimeError):
             aubio.pitchshift("default", -123)
-class aubio_pitchshift_testruns(TestCase):
+class Test_aubio_pitchshift_testruns(object):
-    @params(
+    run_args = ['mode', 'pitchscale', 'hop_size', 'samplerate']
+    run_values = [
             ("default",     1.2,  128,  44100),
             ("crispness:0", 0.43,  64,   8000),
             ("crispness:3", 0.53, 256,   8000),
             ("crispness:3", 1.53, 512,   8000),
             ("crispness:6", 2.3, 4096, 192000),
-            )
+            ]
+    @parametrize(run_args, run_values)
     def test_run_with_params(self, mode, pitchscale, hop_size, samplerate):
         try:
             self.o = aubio.pitchshift(mode, pitchscale, hop_size, samplerate)
             read += len(vec)
 if __name__ == '__main__':
-    from nose2 import main
+    from _tools import run_module_suite
-    main()
+    run_module_suite()

Keyboard shortcuts