#! /usr/bin/env python
-from numpy.testing import TestCase, assert_equal, assert_almost_equal
+from nose2 import main
from nose2.tools import params
+from numpy.testing import TestCase
from aubio import fvec, source, sink
-from numpy import array
-from utils import list_all_sounds, get_tmp_sink_path, del_tmp_sink_path
+from .utils import list_all_sounds, get_tmp_sink_path, del_tmp_sink_path
+
+import warnings
+warnings.filterwarnings('ignore', category=UserWarning, append=True)
list_of_sounds = list_all_sounds('sounds')
samplerates = [0, 44100, 8000, 32000]
if not len(list_of_sounds):
self.skipTest('add some sound files in \'python/tests/sounds\'')
+ def test_wrong_filename(self):
+ with self.assertRaises(RuntimeError):
+ sink('')
+
+ def test_wrong_samplerate(self):
+ with self.assertRaises(RuntimeError):
+ sink(get_tmp_sink_path(), -1)
+
+ def test_wrong_samplerate_too_large(self):
+ with self.assertRaises(RuntimeError):
+ sink(get_tmp_sink_path(), 1536001, 2)
+
+ def test_wrong_channels(self):
+ with self.assertRaises(RuntimeError):
+ sink(get_tmp_sink_path(), 44100, -1)
+
+ def test_wrong_channels_too_large(self):
+ with self.assertRaises(RuntimeError):
+ sink(get_tmp_sink_path(), 44100, 202020)
+
def test_many_sinks(self):
from tempfile import mkdtemp
import os.path
g = sink(path, 0)
sink_list.append(g)
write = 32
- for n in range(200):
+ for _ in range(200):
vec = fvec(write)
g(vec, write)
g.close()
try:
f = source(path, samplerate, hop_size)
except RuntimeError as e:
- self.skipTest('failed opening with hop_s = {:d}, samplerate = {:d} ({:s})'.format(hop_size, samplerate, e))
+ self.skipTest('failed opening with hop_s = {:d}, samplerate = {:d} ({:s})'.format(hop_size, samplerate, str(e)))
if samplerate == 0: samplerate = f.samplerate
sink_path = get_tmp_sink_path()
g = sink(sink_path, samplerate)
try:
f = source(path, samplerate, hop_size)
except RuntimeError as e:
- self.skipTest('failed opening with hop_s = {:d}, samplerate = {:d} ({:s})'.format(hop_size, samplerate, e))
+ self.skipTest('failed opening with hop_s = {:d}, samplerate = {:d} ({:s})'.format(hop_size, samplerate, str(e)))
if samplerate == 0: samplerate = f.samplerate
sink_path = get_tmp_sink_path()
g = sink(sink_path, samplerate, channels = f.channels)
g.close()
del_tmp_sink_path(sink_path)
+ def test_read_with(self):
+ sink_path =get_tmp_sink_path()
+ vec = fvec(128)
+ with sink(sink_path, samplerate) as g:
+ for i in range(10):
+ g(vec, 128)
+
if __name__ == '__main__':
- from nose2 import main
main()