sigrid/aubio

python/tests/__init__.py+0−1

@@ -1 +0,0 @@
1

python/tests/_tools.py+41−0

@@ -0,0 +1,41 @@
	1	"""
	2	This file imports test methods from different testing modules, in this
	3	order:
	4
	5	- try importing 'pytest'
	6	- if it fails, fallback to 'numpy.testing'
	7
	8	Nose2 support was removed because of lacking assertWarns on py2.7.
	9
	10	"""
	11
	12	import sys
	13
	14	_has_pytest = False
	15
	16	# check if we have pytest
	17	try:
	18	import pytest
	19	parametrize = pytest.mark.parametrize
	20	assert_raises = pytest.raises
	21	assert_warns = pytest.warns
	22	skipTest = pytest.skip
	23	_has_pytest = True
	24	def run_module_suite():
	25	import sys, pytest
	26	pytest.main(sys.argv)
	27	except:
	28	pass
	29
	30	# otherwise fallback on numpy.testing
	31	if not _has_pytest:
	32	from numpy.testing import dec, assert_raises, assert_warns
	33	from numpy.testing import SkipTest
	34	parametrize = dec.parametrize
	35	def skipTest(msg):
	36	raise SkipTest(msg)
	37	from numpy.testing import run_module_suite
	38
	39	# always use numpy's assert_equal
	40	import numpy
	41	assert_equal = numpy.testing.assert_equal

python/tests/run_all_tests+0−5

@@ -1,5 +0,0 @@
1	#! /usr/bin/env python
2
3	if __name__ == '__main__':
4	import nose2.main
5	nose2.discover()

python/tests/test_aubio.py+1−2

 #! /usr/bin/env python
-from unittest import main
 from numpy.testing import TestCase
 class aubiomodule_test_case(TestCase):
         self.assertEqual('0', aubio.version[0])
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_aubio_cmd.py+2−2

 #! /usr/bin/env python
-import aubio.cmd
-from nose2 import main
 from numpy.testing import TestCase
+import aubio.cmd
 class aubio_cmd(TestCase):
                 "3200\t")
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_aubio_cut.py+1−1

 #! /usr/bin/env python
 import aubio.cut
-from nose2 import main
 from numpy.testing import TestCase
 class aubio_cut(TestCase):
         assert self.a_parser.parse_args(['-v']).verbose
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_cvec.py+1−1

 #! /usr/bin/env python
-from unittest import main
 import numpy as np
 from numpy.testing import TestCase, assert_equal
 from aubio import cvec, fvec, float_type
             a.norm = np.zeros((512//2+1, 2), dtype = float_type)
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_dct.py+4−0

@@ -66,3 +66,7 @@ class aubio_dct(TestCase): · 65 unmodified lines
66	aubio.dct(size)	66	aubio.dct(size)
67	except AssertionError:	67	except AssertionError:
68	self.skipTest('creating aubio.dct with size %d did not fail' % size)	68	self.skipTest('creating aubio.dct with size %d did not fail' % size)
		69
		70	if __name__ == '__main__':
		71	from unittest import main
		72	main()

python/tests/test_fft.py+1−1

 #! /usr/bin/env python
-from unittest import main
 from numpy.testing import TestCase
 from numpy.testing import assert_equal, assert_almost_equal
 import numpy as np
             fft(win_s)
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_filter.py+2−2

 #! /usr/bin/env python
-from unittest import main
 from numpy.testing import TestCase, assert_equal, assert_almost_equal
 from aubio import fvec, digital_filter
-from .utils import array_from_text_file
+from utils import array_from_text_file
 class aubio_filter_test_case(TestCase):
             digital_filter(-1)
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_filterbank.py+3−3

@@ -4,7 +4,7 @@ import numpy as np · 3 unmodified lines
4	from numpy.testing import TestCase, assert_equal, assert_almost_equal	4	from numpy.testing import TestCase, assert_equal, assert_almost_equal
5		5
6	from aubio import cvec, filterbank, float_type	6	from aubio import cvec, filterbank, float_type
7	from .utils import array_from_text_file	7	from utils import array_from_text_file
8		8
9	class aubio_filterbank_test_case(TestCase):	9	class aubio_filterbank_test_case(TestCase):
10		10
@@ -87,5 +87,5 @@ class aubio_filterbank_wrong_values(TestCase): · 76 unmodified lines
87	f(cvec(256))	87	f(cvec(256))
88		88
89	if __name__ == '__main__':	89	if __name__ == '__main__':
90	import nose2	90	from unittest import main
91	nose2.main()	91	main()

python/tests/test_filterbank_mel.py+14−15

 import numpy as np
 from numpy.testing import TestCase, assert_equal, assert_almost_equal
+from _tools import assert_warns
 from aubio import fvec, cvec, filterbank, float_type
-import warnings
-warnings.filterwarnings('ignore', category=UserWarning, append=True)
 class aubio_filterbank_mel_test_case(TestCase):
     def test_slaney(self):
         samplerate = 22050
         freq_list = [0, samplerate//4, samplerate // 2 + 1]
         f = filterbank(len(freq_list)-2, 1024)
-        # TODO add assert_warns
+        with assert_warns(UserWarning):
-        f.set_triangle_bands(fvec(freq_list), samplerate)
+            f.set_triangle_bands(fvec(freq_list), samplerate)
     def test_triangle_freqs_with_not_enough_filters(self):
         """make sure set_triangle_bands warns when not enough filters"""
         samplerate = 22050
         freq_list = [0, 100, 1000, 4000, 8000, 10000]
         f = filterbank(len(freq_list)-3, 1024)
-        # TODO add assert_warns
+        with assert_warns(UserWarning):
-        f.set_triangle_bands(fvec(freq_list), samplerate)
+            f.set_triangle_bands(fvec(freq_list), samplerate)
     def test_triangle_freqs_with_too_many_filters(self):
         """make sure set_triangle_bands warns when too many filters"""
         samplerate = 22050
         freq_list = [0, 100, 1000, 4000, 8000, 10000]
         f = filterbank(len(freq_list)-1, 1024)
-        # TODO add assert_warns
+        with assert_warns(UserWarning):
-        f.set_triangle_bands(fvec(freq_list), samplerate)
+            f.set_triangle_bands(fvec(freq_list), samplerate)
     def test_triangle_freqs_with_double_value(self):
         """make sure set_triangle_bands works with 2 duplicate freqs"""
         samplerate = 22050
         freq_list = [0, 100, 1000, 4000, 4000, 4000, 10000]
         f = filterbank(len(freq_list)-2, 1024)
-        # TODO add assert_warns
+        with assert_warns(UserWarning):
-        f.set_triangle_bands(fvec(freq_list), samplerate)
+            f.set_triangle_bands(fvec(freq_list), samplerate)
     def test_triangle_freqs_with_triple(self):
         """make sure set_triangle_bands works with 3 duplicate freqs"""
         samplerate = 22050
         freq_list = [0, 100, 1000, 4000, 4000, 4000, 10000]
         f = filterbank(len(freq_list)-2, 1024)
-        # TODO add assert_warns
+        with assert_warns(UserWarning):
-        f.set_triangle_bands(fvec(freq_list), samplerate)
+            f.set_triangle_bands(fvec(freq_list), samplerate)
     def test_triangle_freqs_without_norm(self):
         """make sure set_triangle_bands works without """
 if __name__ == '__main__':
-    import nose2
+    from unittest import main
-    nose2.main()
+    main()

python/tests/test_fvec.py+1−1

 #! /usr/bin/env python
-from unittest import main
 import numpy as np
 from numpy.testing import TestCase, assert_equal, assert_almost_equal
 from aubio import fvec, zero_crossing_rate, alpha_norm, min_removal
         del c
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_fvec_shift.py+1−1

@@ -30,6 +30,6 @@ class aubio_shift_test_case(TestCase): · 29 unmodified lines
30	def test_can_shift_fvec_odd(self):	30	def test_can_shift_fvec_odd(self):
31	self.run_shift_ishift(7)	31	self.run_shift_ishift(7)
32		32
33	from unittest import main
34	if __name__ == '__main__':	33	if __name__ == '__main__':
		34	from unittest import main
35	main()	35	main()

python/tests/test_hztomel.py+11−8

 from unittest import main
 from numpy.testing import TestCase
 from numpy.testing import assert_equal, assert_almost_equal
+from _tools import assert_warns
 import numpy as np
 import aubio
             assert_almost_equal(meltohz(hztomel(f)) - f, 0, decimal=1)
     def test_meltohz_negative(self):
-        # TODO add assert_warns
+        with assert_warns(UserWarning):
-        assert_equal(meltohz(-1), 0)
+            assert_equal(meltohz(-1), 0)
     def test_hztomel_negative(self):
-        # TODO add assert_warns
+        with assert_warns(UserWarning):
-        assert_equal(hztomel(-1), 0)
+            assert_equal(hztomel(-1), 0)
 class aubio_hztomel_htk_test_case(TestCase):
         assert_almost_equal(hztomel(6300, htk=True), 2595., decimal=1)
     def test_meltohz_negative(self):
-        # TODO add assert_warns
+        with assert_warns(UserWarning):
-        assert_equal(meltohz(-1, htk=True), 0)
+            assert_equal(meltohz(-1, htk=True), 0)
         assert_almost_equal(meltohz(2000, htk=True), 3428.7, decimal=1)
         assert_almost_equal(meltohz(1000, htk=True), 1000., decimal=1)
     def test_hztomel_negative(self):
-        # TODO add assert_warns
+        with assert_warns(UserWarning):
-        assert_equal(hztomel(-1, htk=True), 0)
+            assert_equal(meltohz(-1, htk=True), 0)
+        with assert_warns(UserWarning):
+            assert_equal(hztomel(-1, htk=True), 0)
         assert_almost_equal(hztomel(1000, htk=True), 1000., decimal=1)
     def test_hztomel_htk(self):

python/tests/test_mathutils.py+1−1

 #! /usr/bin/env python
-from unittest import main
 from numpy.testing import TestCase, assert_equal
 from numpy import array, arange, isnan, isinf
 from aubio import bintomidi, miditobin, freqtobin, bintofreq, freqtomidi, miditofreq
         assert_equal ( array(b) < 0, False )
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_mfcc.py+17−21

 #! /usr/bin/env python
-from nose2 import main
+from _tools import parametrize, assert_raises
-from nose2.tools import params
 from numpy import random, count_nonzero
 from numpy.testing import TestCase
 from aubio import mfcc, cvec, float_type
 new_params = ['buf_size', 'n_filters', 'n_coeffs', 'samplerate']
 new_deflts = [1024, 40, 13, 44100]
-class aubio_mfcc(TestCase):
+class Test_aubio_mfcc(object):
-    def setUp(self):
+    members_args = 'name'
-        self.o = mfcc()
-    def test_default_creation(self):
+    @parametrize(members_args, new_params)
-        pass
-    def test_delete(self):
-        del self.o
-    @params(*new_params)
     def test_read_only_member(self, name):
-        o = self.o
+        o = mfcc()
-        with self.assertRaises((TypeError, AttributeError)):
+        with assert_raises((TypeError, AttributeError)):
             setattr(o, name, 0)
-    @params(*zip(new_params, new_deflts))
+    @parametrize('name, expected', zip(new_params, new_deflts))
     def test_default_param(self, name, expected):
         """ test mfcc.{:s} = {:d} """.format(name, expected)
-        o = self.o
+        o = mfcc()
-        self.assertEqual( getattr(o, name), expected)
+        assert getattr(o, name) == expected
 class aubio_mfcc_wrong_params(TestCase):
         #print coeffs
-class aubio_mfcc_all_parameters(TestCase):
+class Test_aubio_mfcc_all_parameters(object):
-    @params(
+    run_values = [
             (2048, 40, 13, 44100),
             (1024, 40, 13, 44100),
             (512, 40, 13, 44100),
             #(1024, 30, 20, 44100),
             (1024, 40, 40, 44100),
             (1024, 40, 3, 44100),
-            )
+            ]
+    run_args = ['buf_size', 'n_filters', 'n_coeffs', 'samplerate']
+    @parametrize(run_args, run_values)
     def test_run_with_params(self, buf_size, n_filters, n_coeffs, samplerate):
         " check mfcc can run with reasonable parameters "
         o = mfcc(buf_size, n_filters, n_coeffs, samplerate)
         assert m.get_scale() == 1
 if __name__ == '__main__':
-    main()
+    from _tools import run_module_suite
+    run_module_suite()

python/tests/test_midi2note.py+11−12

 # -*- coding: utf-8 -*-
 from aubio import midi2note
-from nose2.tools import params
+from _tools import parametrize, assert_raises
-import unittest
 list_of_known_midis = (
         ( 0, 'C-1' ),
         ( 127, 'G9' ),
         )
-class midi2note_good_values(unittest.TestCase):
+class Test_midi2note_good_values(object):
-    @params(*list_of_known_midis)
+    @parametrize('midi, note', list_of_known_midis)
     def test_midi2note_known_values(self, midi, note):
         " known values are correctly converted "
-        self.assertEqual ( midi2note(midi), note )
+        assert midi2note(midi) == (note)
-class midi2note_wrong_values(unittest.TestCase):
+class Test_midi2note_wrong_values(object):
     def test_midi2note_negative_value(self):
         " fails when passed a negative value "
-        self.assertRaises(ValueError, midi2note, -2)
+        assert_raises(ValueError, midi2note, -2)
     def test_midi2note_large(self):
         " fails when passed a value greater than 127 "
-        self.assertRaises(ValueError, midi2note, 128)
+        assert_raises(ValueError, midi2note, 128)
     def test_midi2note_floating_value(self):
         " fails when passed a floating point "
-        self.assertRaises(TypeError, midi2note, 69.2)
+        assert_raises(TypeError, midi2note, 69.2)
     def test_midi2note_character_value(self):
         " fails when passed a value that can not be transformed to integer "
-        self.assertRaises(TypeError, midi2note, "a")
+        assert_raises(TypeError, midi2note, "a")
 if __name__ == '__main__':
-    import nose2
+    from _tools import run_module_suite
-    nose2.main()
+    run_module_suite()

python/tests/test_musicutils.py+2−2

 #! /usr/bin/env python
-from unittest import main
 import numpy as np
 from numpy.testing import TestCase
-from numpy.testing.utils import assert_equal, assert_almost_equal
+from numpy.testing import assert_equal, assert_almost_equal
 from aubio import window, level_lin, db_spl, silence_detection, level_detection
 from aubio import fvec, float_type
         assert level_detection(ones(1024, dtype = float_type), -70) == 0
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_note2midi.py+24−19

 from __future__ import unicode_literals
 from aubio import note2midi, freq2note, note2freq, float_type
-from nose2.tools import params
+from numpy.testing import TestCase
-import unittest
+from _tools import parametrize, assert_raises, skipTest
 list_of_known_notes = (
         ( 'C-1', 0 ),
         ( '2' ),
         )
-class note2midi_good_values(unittest.TestCase):
+class Test_note2midi_good_values(object):
-    @params(*list_of_known_notes)
+    @parametrize('note, midi', list_of_known_notes)
     def test_note2midi_known_values(self, note, midi):
         " known values are correctly converted "
-        self.assertEqual ( note2midi(note), midi )
+        assert note2midi(note) == midi
-    @params(*list_of_known_notes_with_unicode_issues)
+    @parametrize('note, midi', list_of_known_notes_with_unicode_issues)
     def test_note2midi_known_values_with_unicode_issues(self, note, midi):
-        " known values are correctly converted, unless decoding is expected to fail"
+        " difficult values are correctly converted unless expected failure "
         try:
-            self.assertEqual ( note2midi(note), midi )
+            assert note2midi(note) == midi
         except UnicodeEncodeError as e:
+            # platforms with decoding failures include:
+            # - osx: python <= 2.7.10
+            # - win: python <= 2.7.12
             import sys
-            strfmt = "len(u'\\U0001D12A') != 1, excpected decoding failure | {:s} | {:s} {:s}"
+            strmsg = "len(u'\\U0001D12A') != 1, expected decoding failure"
-            strres = strfmt.format(e, sys.platform, sys.version)
+            strmsg += " | upgrade to Python 3 to fix"
-            # happens with: darwin 2.7.10, windows 2.7.12
+            strmsg += " | {:s} | {:s} {:s}"
             if len('\U0001D12A') != 1 and sys.version[0] == '2':
-                self.skipTest(strres + " | upgrade to Python 3 to fix")
+                skipTest(strmsg.format(repr(e), sys.platform, sys.version))
             else:
                 raise
-class note2midi_wrong_values(unittest.TestCase):
+class note2midi_wrong_values(TestCase):
     def test_note2midi_missing_octave(self):
         " fails when passed only one character"
         " fails when passed a note with a note name longer than expected"
         self.assertRaises(ValueError, note2midi, 'CB+-3')
-    @params(*list_of_unknown_notes)
+class Test_note2midi_unknown_values(object):
+    @parametrize('note', list_of_unknown_notes)
     def test_note2midi_unknown_values(self, note):
         " unknown values throw out an error "
-        self.assertRaises(ValueError, note2midi, note)
+        assert_raises(ValueError, note2midi, note)
-class freq2note_simple_test(unittest.TestCase):
+class freq2note_simple_test(TestCase):
     def test_freq2note_above(self):
         " make sure freq2note(441) == A4 "
         " make sure freq2note(439) == A4 "
         self.assertEqual("A4", freq2note(439))
-class note2freq_simple_test(unittest.TestCase):
+class note2freq_simple_test(TestCase):
     def test_note2freq(self):
         " make sure note2freq('A3') == 220"
             self.assertLess(abs(note2freq("A4")-440), 1.e-12)
 if __name__ == '__main__':
-    import nose2
+    from _tools import run_module_suite
-    nose2.main()
+    run_module_suite()

python/tests/test_notes.py+6−7

 #! /usr/bin/env python
-from unittest import main
 from numpy.testing import TestCase, assert_equal, assert_almost_equal
-from aubio import notes
+from aubio import notes, source
+import numpy as np
+from utils import list_all_sounds
+list_of_sounds = list_all_sounds('sounds')
 AUBIO_DEFAULT_NOTES_SILENCE = -70.
 AUBIO_DEFAULT_NOTES_RELEASE_DROP = 10.
         with self.assertRaises(ValueError):
             self.o.set_release_drop(val)
-from .utils import list_all_sounds
-list_of_sounds = list_all_sounds('sounds')
 class aubio_notes_sinewave(TestCase):
     def analyze_file(self, filepath, samplerate=0):
-        from aubio import source
-        import numpy as np
         win_s = 512 # fft size
         hop_s = 256 # hop size
                 assert_equal (results[0][1], [69, 123, -1])
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_onset.py+1−1

@@ -1,6 +1,5 @@
1	#! /usr/bin/env python	1	#! /usr/bin/env python
2		2
3	from unittest import main
4	from numpy.testing import TestCase, assert_equal, assert_almost_equal	3	from numpy.testing import TestCase, assert_equal, assert_almost_equal
5	from aubio import onset, fvec	4	from aubio import onset, fvec
6		5
@@ -116,4 +115,5 @@ class aubio_onset_coverate(TestCase): · 109 unmodified lines
116		115
117		116
118	if __name__ == '__main__':	117	if __name__ == '__main__':
		118	from unittest import main
119	main()	119	main()

python/tests/test_phasevoc.py+13−8

 #! /usr/bin/env python
 from numpy.testing import TestCase, assert_equal, assert_array_less
+from _tools import parametrize
 from aubio import fvec, cvec, pvoc, float_type
-from nose2 import main
-from nose2.tools import params
 import numpy as np
 if float_type == 'float32':
 def create_noise(hop_s):
     return np.random.rand(hop_s).astype(float_type) * 2. - 1.
-class aubio_pvoc_test_case(TestCase):
+class Test_aubio_pvoc_test_case(object):
     """ pvoc object test case """
     def test_members_automatic_sizes_default(self):
             r = f.rdo(s)
             assert_equal ( t, 0.)
-    @params(
+    resynth_noise_args = "hop_s, ratio"
+    resynth_noise_values = [
             ( 256, 8),
             ( 256, 4),
             ( 256, 2),
             (8192, 8),
             (8192, 4),
             (8192, 2),
-            )
+            ]
+    @parametrize(resynth_noise_args, resynth_noise_values)
     def test_resynth_steps_noise(self, hop_s, ratio):
         """ check the resynthesis of a random signal is correct """
         sigin = create_noise(hop_s)
         self.reconstruction(sigin, hop_s, ratio)
-    @params(
+    resynth_sine_args = "samplerate, hop_s, ratio, freq"
+    resynth_sine_values = [
             (44100,  256, 8,   441),
             (44100,  256, 4,  1203),
             (44100,  256, 2,  3045),
             (22050,  256, 8,   445),
             (96000, 1024, 8, 47000),
             (96000, 1024, 8,    20),
-            )
+            ]
+    @parametrize(resynth_sine_args, resynth_sine_values)
     def test_resynth_steps_sine(self, samplerate, hop_s, ratio, freq):
         """ check the resynthesis of a sine is correct """
         sigin = create_sine(hop_s, freq, samplerate)
             self.skipTest('creating aubio.pvoc with size %d did not fail' % win_s)
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_pitch.py+5−5

 #! /usr/bin/env python
-from unittest import TestCase, main
+from numpy.testing import TestCase, assert_equal
-from numpy.testing import assert_equal
 from numpy import sin, arange, mean, median, isnan, pi
 from aubio import fvec, pitch, freqtomidi, float_type
 for algo in pitch_algorithms:
     for mode in signal_modes:
-        test_method = create_test (algo, mode)
+        _test_method = create_test (algo, mode)
-        test_method.__name__ = 'test_pitch_%s_%d_%d_%dHz_sin_%.0f' % ( algo,
+        _test_method.__name__ = 'test_pitch_%s_%d_%d_%dHz_sin_%.0f' % ( algo,
                 mode[0], mode[1], mode[2], mode[3] )
-        setattr (aubio_pitch_Sinusoid, test_method.__name__, test_method)
+        setattr (aubio_pitch_Sinusoid, _test_method.__name__, _test_method)
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_sink.py+16−22

@@ -1,13 +1,9 @@
1	#! /usr/bin/env python	1	#! /usr/bin/env python
2		2
3	from nose2 import main
4	from nose2.tools import params
5	from numpy.testing import TestCase	3	from numpy.testing import TestCase
6	from aubio import fvec, source, sink	4	from aubio import fvec, source, sink
7	from .utils import list_all_sounds, get_tmp_sink_path, del_tmp_sink_path	5	from utils import list_all_sounds, get_tmp_sink_path, del_tmp_sink_path
8		6	from _tools import parametrize, skipTest, assert_raises
9	import warnings
10	warnings.filterwarnings('ignore', category=UserWarning, append=True)
11		7
12	list_of_sounds = list_all_sounds('sounds')	8	list_of_sounds = list_all_sounds('sounds')
13	samplerates = [0, 44100, 8000, 32000]	9	samplerates = [0, 44100, 8000, 32000]
@@ -23,30 +19,26 @@ for soundfile in list_of_sounds: · 9 unmodified lines
23	for samplerate in samplerates:	19	for samplerate in samplerates:
24	all_params.append((hop_size, samplerate, soundfile))	20	all_params.append((hop_size, samplerate, soundfile))
25		21
26	class aubio_sink_test_case(TestCase):	22	class Test_aubio_sink(object):
27
28	def setUp(self):
29	if not len(list_of_sounds):
30	self.skipTest('add some sound files in \'python/tests/sounds\'')
31		23
32	def test_wrong_filename(self):	24	def test_wrong_filename(self):
33	with self.assertRaises(RuntimeError):	25	with assert_raises(RuntimeError):
34	sink('')	26	sink('')
35		27
36	def test_wrong_samplerate(self):	28	def test_wrong_samplerate(self):
37	with self.assertRaises(RuntimeError):	29	with assert_raises(RuntimeError):
38	sink(get_tmp_sink_path(), -1)	30	sink(get_tmp_sink_path(), -1)
39		31
40	def test_wrong_samplerate_too_large(self):	32	def test_wrong_samplerate_too_large(self):
41	with self.assertRaises(RuntimeError):	33	with assert_raises(RuntimeError):
42	sink(get_tmp_sink_path(), 1536001, 2)	34	sink(get_tmp_sink_path(), 1536001, 2)
43		35
44	def test_wrong_channels(self):	36	def test_wrong_channels(self):
45	with self.assertRaises(RuntimeError):	37	with assert_raises(RuntimeError):
46	sink(get_tmp_sink_path(), 44100, -1)	38	sink(get_tmp_sink_path(), 44100, -1)
47		39
48	def test_wrong_channels_too_large(self):	40	def test_wrong_channels_too_large(self):
49	with self.assertRaises(RuntimeError):	41	with assert_raises(RuntimeError):
50	sink(get_tmp_sink_path(), 44100, 202020)	42	sink(get_tmp_sink_path(), 44100, 202020)
51		43
52	def test_many_sinks(self):	44	def test_many_sinks(self):
@@ -66,13 +58,13 @@ class aubio_sink_test_case(TestCase): · 13 unmodified lines
66	g.close()	58	g.close()
67	shutil.rmtree(tmpdir)	59	shutil.rmtree(tmpdir)
68		60
69	@params(*all_params)	61	@parametrize('hop_size, samplerate, path', all_params)
70	def test_read_and_write(self, hop_size, samplerate, path):	62	def test_read_and_write(self, hop_size, samplerate, path):
71
72	try:	63	try:
73	f = source(path, samplerate, hop_size)	64	f = source(path, samplerate, hop_size)
74	except RuntimeError as e:	65	except RuntimeError as e:
75	self.skipTest('failed opening with hop_s = {:d}, samplerate = {:d} ({:s})'.format(hop_size, samplerate, str(e)))	66	err_msg = '{:s} (hop_s = {:d}, samplerate = {:d})'
		67	skipTest(err_msg.format(str(e), hop_size, samplerate))
76	if samplerate == 0: samplerate = f.samplerate	68	if samplerate == 0: samplerate = f.samplerate
77	sink_path = get_tmp_sink_path()	69	sink_path = get_tmp_sink_path()
78	g = sink(sink_path, samplerate)	70	g = sink(sink_path, samplerate)
@@ -84,12 +76,13 @@ class aubio_sink_test_case(TestCase): · 5 unmodified lines
84	if read < f.hop_size: break	76	if read < f.hop_size: break
85	del_tmp_sink_path(sink_path)	77	del_tmp_sink_path(sink_path)
86		78
87	@params(*all_params)	79	@parametrize('hop_size, samplerate, path', all_params)
88	def test_read_and_write_multi(self, hop_size, samplerate, path):	80	def test_read_and_write_multi(self, hop_size, samplerate, path):
89	try:	81	try:
90	f = source(path, samplerate, hop_size)	82	f = source(path, samplerate, hop_size)
91	except RuntimeError as e:	83	except RuntimeError as e:
92	self.skipTest('failed opening with hop_s = {:d}, samplerate = {:d} ({:s})'.format(hop_size, samplerate, str(e)))	84	err_msg = '{:s} (hop_s = {:d}, samplerate = {:d})'
		85	skipTest(err_msg.format(str(e), hop_size, samplerate))
93	if samplerate == 0: samplerate = f.samplerate	86	if samplerate == 0: samplerate = f.samplerate
94	sink_path = get_tmp_sink_path()	87	sink_path = get_tmp_sink_path()
95	g = sink(sink_path, samplerate, channels = f.channels)	88	g = sink(sink_path, samplerate, channels = f.channels)
@@ -125,4 +118,5 @@ class aubio_sink_test_case(TestCase): · 29 unmodified lines
125	g(vec, 128)	118	g(vec, 128)
126		119
127	if __name__ == '__main__':	120	if __name__ == '__main__':
128	main()	121	from _tools import run_module_suite
		122	run_module_suite()

python/tests/test_slicing.py+3−3

 #! /usr/bin/env python
-from unittest import main
 from numpy.testing import TestCase, assert_equal
 from aubio import slice_source_at_stamps
-from .utils import count_files_in_directory, get_default_test_sound
+from utils import count_files_in_directory, get_default_test_sound
-from .utils import count_samples_in_directory, count_samples_in_file
+from utils import count_samples_in_directory, count_samples_in_file
 import tempfile
 import shutil
         shutil.rmtree(self.output_dir)
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_source.py+56−53

@@ -1,19 +1,17 @@
1	#! /usr/bin/env python	1	#! /usr/bin/env python
2		2
3	from nose2 import main	3
4	from nose2.tools import params
5	from numpy.testing import TestCase, assert_equal	4	from numpy.testing import TestCase, assert_equal
6	from aubio import source	5	from aubio import source
7	from .utils import list_all_sounds	6	from utils import list_all_sounds
8		7	import unittest
9	import warnings	8	from _tools import parametrize, assert_raises, assert_equal, skipTest
10	warnings.filterwarnings('ignore', category=UserWarning, append=True)
11		9
12	list_of_sounds = list_all_sounds('sounds')	10	list_of_sounds = list_all_sounds('sounds')
13	samplerates = [0, 44100, 8000, 32000]	11	samplerates = [0, 44100, 8000, 32000]
14	hop_sizes = [512, 1024, 64]	12	hop_sizes = [512, 1024, 64]
15		13
16	path = None	14	default_test_sound = len(list_of_sounds) and list_of_sounds[0] or None
17		15
18	all_params = []	16	all_params = []
19	for soundfile in list_of_sounds:	17	for soundfile in list_of_sounds:
@@ -21,24 +19,20 @@ for soundfile in list_of_sounds: · 1 unmodified line
21	for samplerate in samplerates:	19	for samplerate in samplerates:
22	all_params.append((hop_size, samplerate, soundfile))	20	all_params.append((hop_size, samplerate, soundfile))
23		21
		22	no_sounds_msg = "no test sounds, add some in 'python/tests/sounds/'!"
24		23
25	class aubio_source_test_case_base(TestCase):	24	_debug = False
26
27	def setUp(self):
28	if not len(list_of_sounds):
29	self.skipTest('add some sound files in \'python/tests/sounds\'')
30	self.default_test_sound = list_of_sounds[0]
31		25
32	class aubio_source_test_case(aubio_source_test_case_base):	26	class Test_aubio_source_test_case(object):
33		27
34	@params(*list_of_sounds)	28	@parametrize('filename', list_of_sounds)
35	def test_close_file(self, filename):	29	def test_close_file(self, filename):
36	samplerate = 0 # use native samplerate	30	samplerate = 0 # use native samplerate
37	hop_size = 256	31	hop_size = 256
38	f = source(filename, samplerate, hop_size)	32	f = source(filename, samplerate, hop_size)
39	f.close()	33	f.close()
40		34
41	@params(*list_of_sounds)	35	@parametrize('filename', list_of_sounds)
42	def test_close_file_twice(self, filename):	36	def test_close_file_twice(self, filename):
43	samplerate = 0 # use native samplerate	37	samplerate = 0 # use native samplerate
44	hop_size = 256	38	hop_size = 256
@@ -46,7 +40,7 @@ class aubio_source_test_case(aubio_source_test_case_base): · 1 unmodified line
46	f.close()	40	f.close()
47	f.close()	41	f.close()
48		42
49	class aubio_source_read_test_case(aubio_source_test_case_base):	43	class Test_aubio_source_read(object):
50		44
51	def read_from_source(self, f):	45	def read_from_source(self, f):
52	total_frames = 0	46	total_frames = 0
@@ -56,47 +50,51 @@ class aubio_source_read_test_case(aubio_source_test_case_base): · 3 unmodified lines
56	if read < f.hop_size:	50	if read < f.hop_size:
57	assert_equal(samples[read:], 0)	51	assert_equal(samples[read:], 0)
58	break	52	break
59	#result_str = "read {:.2f}s ({:d} frames in {:d} blocks at {:d}Hz) from {:s}"	53	if _debug:
60	#result_params = total_frames / float(f.samplerate), total_frames, total_frames//f.hop_size, f.samplerate, f.uri	54	result_str = "read {:.2f}s ({:d} frames"
61	#print (result_str.format(*result_params))	55	result_str += " in {:d} blocks at {:d}Hz) from {:s}"
		56	result_params = total_frames / float(f.samplerate), total_frames, \
		57	total_frames//f.hop_size, f.samplerate, f.uri
		58	print (result_str.format(*result_params))
62	return total_frames	59	return total_frames
63		60
64	@params(*all_params)	61	@parametrize('hop_size, samplerate, soundfile', all_params)
65	def test_samplerate_hopsize(self, hop_size, samplerate, soundfile):	62	def test_samplerate_hopsize(self, hop_size, samplerate, soundfile):
66	try:	63	try:
67	f = source(soundfile, samplerate, hop_size)	64	f = source(soundfile, samplerate, hop_size)
68	except RuntimeError as e:	65	except RuntimeError as e:
69	self.skipTest('failed opening with hop_s = {:d}, samplerate = {:d} ({:s})'.format(hop_size, samplerate, str(e)))	66	err_msg = 'failed opening with hop_s={:d}, samplerate={:d} ({:s})'
		67	skipTest(err_msg.format(hop_size, samplerate, str(e)))
70	assert f.samplerate != 0	68	assert f.samplerate != 0
71	read_frames = self.read_from_source(f)	69	read_frames = self.read_from_source(f)
72	if 'f_' in soundfile and samplerate == 0:	70	if 'f_' in soundfile and samplerate == 0:
73	import re	71	import re
74	f = re.compile('._\([0:9]f\)_.*')	72	f = re.compile(r'._\([0:9]f\)_.*')
75	match_f = re.findall('([0-9]*)f_', soundfile)	73	match_f = re.findall('([0-9]*)f_', soundfile)
76	if len(match_f) == 1:	74	if len(match_f) == 1:
77	expected_frames = int(match_f[0])	75	expected_frames = int(match_f[0])
78	self.assertEqual(expected_frames, read_frames)	76	assert_equal(expected_frames, read_frames)
79		77
80	@params(*list_of_sounds)	78	@parametrize('p', list_of_sounds)
81	def test_samplerate_none(self, p):	79	def test_samplerate_none(self, p):
82	f = source(p)	80	f = source(p)
83	assert f.samplerate != 0	81	assert f.samplerate != 0
84	self.read_from_source(f)	82	self.read_from_source(f)
85		83
86	@params(*list_of_sounds)	84	@parametrize('p', list_of_sounds)
87	def test_samplerate_0(self, p):	85	def test_samplerate_0(self, p):
88	f = source(p, 0)	86	f = source(p, 0)
89	assert f.samplerate != 0	87	assert f.samplerate != 0
90	self.read_from_source(f)	88	self.read_from_source(f)
91		89
92	@params(*list_of_sounds)	90	@parametrize('p', list_of_sounds)
93	def test_zero_hop_size(self, p):	91	def test_zero_hop_size(self, p):
94	f = source(p, 0, 0)	92	f = source(p, 0, 0)
95	assert f.samplerate != 0	93	assert f.samplerate != 0
96	assert f.hop_size != 0	94	assert f.hop_size != 0
97	self.read_from_source(f)	95	self.read_from_source(f)
98		96
99	@params(*list_of_sounds)	97	@parametrize('p', list_of_sounds)
100	def test_seek_to_half(self, p):	98	def test_seek_to_half(self, p):
101	from random import randint	99	from random import randint
102	f = source(p, 0, 0)	100	f = source(p, 0, 0)
@@ -108,7 +106,7 @@ class aubio_source_read_test_case(aubio_source_test_case_base): · 5 unmodified lines
108	b = self.read_from_source(f)	106	b = self.read_from_source(f)
109	assert a == b + c	107	assert a == b + c
110		108
111	@params(*list_of_sounds)	109	@parametrize('p', list_of_sounds)
112	def test_duration(self, p):	110	def test_duration(self, p):
113	total_frames = 0	111	total_frames = 0
114	f = source(p)	112	f = source(p)
@@ -117,43 +115,44 @@ class aubio_source_read_test_case(aubio_source_test_case_base): · 2 unmodified lines
117	_, read = f()	115	_, read = f()
118	total_frames += read	116	total_frames += read
119	if read < f.hop_size: break	117	if read < f.hop_size: break
120	self.assertEqual(duration, total_frames)	118	assert_equal (duration, total_frames)
121		119
122		120
123	class aubio_source_test_wrong_params(TestCase):	121	class Test_aubio_source_wrong_params(object):
124		122
125	def test_wrong_file(self):	123	def test_wrong_file(self):
126	with self.assertRaises(RuntimeError):	124	with assert_raises(RuntimeError):
127	source('path_to/unexisting file.mp3')	125	source('path_to/unexisting file.mp3')
128		126
129	class aubio_source_test_wrong_params_with_file(aubio_source_test_case_base):	127	@unittest.skipIf(default_test_sound is None, no_sounds_msg)
		128	class Test_aubio_source_wrong_params_with_file(TestCase):
130		129
131	def test_wrong_samplerate(self):	130	def test_wrong_samplerate(self):
132	with self.assertRaises(ValueError):	131	with assert_raises(ValueError):
133	source(self.default_test_sound, -1)	132	source(default_test_sound, -1)
134		133
135	def test_wrong_hop_size(self):	134	def test_wrong_hop_size(self):
136	with self.assertRaises(ValueError):	135	with assert_raises(ValueError):
137	source(self.default_test_sound, 0, -1)	136	source(default_test_sound, 0, -1)
138		137
139	def test_wrong_channels(self):	138	def test_wrong_channels(self):
140	with self.assertRaises(ValueError):	139	with assert_raises(ValueError):
141	source(self.default_test_sound, 0, 0, -1)	140	source(default_test_sound, 0, 0, -1)
142		141
143	def test_wrong_seek(self):	142	def test_wrong_seek(self):
144	f = source(self.default_test_sound)	143	f = source(default_test_sound)
145	with self.assertRaises(ValueError):	144	with assert_raises(ValueError):
146	f.seek(-1)	145	f.seek(-1)
147		146
148	def test_wrong_seek_too_large(self):	147	def test_wrong_seek_too_large(self):
149	f = source(self.default_test_sound)	148	f = source(default_test_sound)
150	try:	149	try:
151	with self.assertRaises(ValueError):	150	with assert_raises(ValueError):
152	f.seek(f.duration + f.samplerate * 10)	151	f.seek(f.duration + f.samplerate * 10)
153	except AssertionError:	152	except:
154	self.skipTest('seeking after end of stream failed raising ValueError')	153	skipTest('seeking after end of stream failed raising ValueError')
155		154
156	class aubio_source_readmulti_test_case(aubio_source_read_test_case):	155	class Test_aubio_source_readmulti(Test_aubio_source_read):
157		156
158	def read_from_source(self, f):	157	def read_from_source(self, f):
159	total_frames = 0	158	total_frames = 0
@@ -163,15 +162,18 @@ class aubio_source_readmulti_test_case(aubio_source_read_test_case): · 3 unmodified lines
163	if read < f.hop_size:	162	if read < f.hop_size:
164	assert_equal(samples[:,read:], 0)	163	assert_equal(samples[:,read:], 0)
165	break	164	break
166	#result_str = "read {:.2f}s ({:d} frames in {:d} channels and {:d} blocks at {:d}Hz) from {:s}"	165	if _debug:
167	#result_params = total_frames / float(f.samplerate), total_frames, f.channels, int(total_frames/f.hop_size), f.samplerate, f.uri	166	result_str = "read {:.2f}s ({:d} frames in {:d} channels"
168	#print (result_str.format(*result_params))	167	result_str += " and {:d} blocks at {:d}Hz) from {:s}"
		168	result_params = total_frames / float(f.samplerate), total_frames, \
		169	f.channels, int(total_frames/f.hop_size), \
		170	f.samplerate, f.uri
		171	print (result_str.format(*result_params))
169	return total_frames	172	return total_frames
170		173
171	class aubio_source_with(aubio_source_test_case_base):	174	class Test_aubio_source_with(object):
172		175
173	#@params(*list_of_sounds)	176	@parametrize('filename', list_of_sounds)
174	@params(*list_of_sounds)
175	def test_read_from_mono(self, filename):	177	def test_read_from_mono(self, filename):
176	total_frames = 0	178	total_frames = 0
177	hop_size = 2048	179	hop_size = 2048
@@ -185,4 +187,5 @@ class aubio_source_with(aubio_source_test_case_base): · 7 unmodified lines
185	assert_equal(total_frames, input_source.duration)	187	assert_equal(total_frames, input_source.duration)
186		188
187	if __name__ == '__main__':	189	if __name__ == '__main__':
188	main()	190	from _tools import run_module_suite
		191	run_module_suite()

python/tests/test_source_channels.py+1−1

@@ -8,7 +8,7 @@ import unittest · 7 unmodified lines
8	import aubio	8	import aubio
9	import numpy as np	9	import numpy as np
10	from numpy.testing import assert_equal	10	from numpy.testing import assert_equal
11	from .utils import get_tmp_sink_path	11	from utils import get_tmp_sink_path
12		12
13	class aubio_source_test_case(unittest.TestCase):	13	class aubio_source_test_case(unittest.TestCase):
14		14

python/tests/test_specdesc.py+1−1

 #! /usr/bin/env python
-from unittest import main
 from numpy.testing import TestCase, assert_equal, assert_almost_equal
 from numpy import random, arange, log, zeros
 from aubio import specdesc, cvec, float_type
             specdesc("unknown", 512)
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/test_zero_crossing_rate.py+1−1

 #! /usr/bin/env python
-from unittest import main
 from numpy.testing import TestCase
 from aubio import fvec, zero_crossing_rate
         self.assertEqual(2./buf_size, zero_crossing_rate(self.vector))
 if __name__ == '__main__':
+    from unittest import main
     main()

python/tests/utils.py+6−12

 DEFAULT_SOUND = '22050Hz_5s_brownnoise.wav'
 def array_from_text_file(filename, dtype = 'float'):
-    filename = os.path.join(os.path.dirname(__file__), filename)
+    realpathname = os.path.join(os.path.dirname(__file__), filename)
-    with open(filename) as f:
+    return np.loadtxt(realpathname, dtype = dtype)
-        lines = f.readlines()
-    return np.array([line.split() for line in lines],
-            dtype = dtype)
 def list_all_sounds(rel_dir):
     datadir = os.path.join(os.path.dirname(__file__), rel_dir)
     try:
         os.unlink(path)
     except WindowsError as e:
-        print("deleting {:s} failed ({:s}), reopening".format(path, repr(e)))
+        # removing the temporary directory sometimes fails on windows
-        with open(path, 'wb') as f:
+        import warnings
-            f.close()
+        errmsg = "failed deleting temporary file {:s} ({:s})"
-        try:
+        warnings.warn(UserWarning(errmsg.format(path, repr(e))))
-            os.unlink(path)
-        except WindowsError as f:
-            print("deleting {:s} failed ({:s}), aborting".format(path, repr(e)))
 def array_from_yaml_file(filename):
     import yaml

Merge branch 'master' into feature/pitchshift

Keyboard shortcuts

General

Issues and pull requests

Issues, pull requests and Inbox

Repository code