source: python/tests/test_musicutils.py @ 58a82d4

feature/autosinkfeature/cnnfeature/cnn_orgfeature/constantqfeature/crepefeature/crepe_orgfeature/pitchshiftfeature/pydocstringsfeature/timestretchfix/ffmpeg5
Last change on this file since 58a82d4 was 0b6d23d, checked in by Paul Brossier <piem@piem.org>, 8 years ago

python/tests: fix most prospect warnings

  • Property mode set to 100755
File size: 2.6 KB
Line 
1#! /usr/bin/env python
2
3from unittest import main
4import numpy as np
5from numpy.testing import TestCase
6from numpy.testing.utils import assert_equal, assert_almost_equal
7from aubio import window, level_lin, db_spl, silence_detection, level_detection
8from aubio import fvec, float_type
9
10class aubio_window(TestCase):
11
12    def test_accept_name_and_size(self):
13        window("default", 1024)
14
15    def test_fail_name_not_string(self):
16        with self.assertRaises(TypeError):
17            window(10, 1024)
18
19    def test_fail_size_not_int(self):
20        with self.assertRaises(TypeError):
21            window("default", "default")
22
23    def test_compute_hanning_1024(self):
24        size = 1024
25        aubio_window = window("hanning", size)
26        numpy_window = .5 - .5 * np.cos(2. * np.pi * np.arange(size) / size)
27        assert_almost_equal(aubio_window, numpy_window)
28
29class aubio_level_lin(TestCase):
30    def test_accept_fvec(self):
31        level_lin(fvec(1024))
32
33    def test_fail_not_fvec(self):
34        with self.assertRaises(ValueError):
35            level_lin("default")
36
37    def test_zeros_is_zeros(self):
38        assert_equal(level_lin(fvec(1024)), 0.)
39
40    def test_minus_ones_is_one(self):
41        assert_equal(level_lin(-np.ones(1024, dtype = float_type)), 1.)
42
43class aubio_db_spl(TestCase):
44    def test_accept_fvec(self):
45        db_spl(fvec(1024))
46
47    def test_fail_not_fvec(self):
48        with self.assertRaises(ValueError):
49            db_spl("default")
50
51    def test_zeros_is_inf(self):
52        assert np.isinf(db_spl(fvec(1024)))
53
54    def test_minus_ones_is_zero(self):
55        assert_equal(db_spl(-np.ones(1024, dtype = float_type)), 0.)
56
57class aubio_silence_detection(TestCase):
58    def test_accept_fvec(self):
59        silence_detection(fvec(1024), -70.)
60
61    def test_fail_not_fvec(self):
62        with self.assertRaises(ValueError):
63            silence_detection("default", -70)
64
65    def test_zeros_is_one(self):
66        assert silence_detection(fvec(1024), -70) == 1
67
68    def test_minus_ones_is_zero(self):
69        from numpy import ones
70        assert silence_detection(ones(1024, dtype = float_type), -70) == 0
71
72class aubio_level_detection(TestCase):
73    def test_accept_fvec(self):
74        level_detection(fvec(1024), -70.)
75
76    def test_fail_not_fvec(self):
77        with self.assertRaises(ValueError):
78            level_detection("default", -70)
79
80    def test_zeros_is_one(self):
81        assert level_detection(fvec(1024), -70) == 1
82
83    def test_minus_ones_is_zero(self):
84        from numpy import ones
85        assert level_detection(ones(1024, dtype = float_type), -70) == 0
86
87if __name__ == '__main__':
88    main()
Note: See TracBrowser for help on using the repository browser.