python/tests: fix most prospect warnings
[aubio.git] / python / tests / test_musicutils.py
1 #! /usr/bin/env python
2
3 from unittest import main
4 import numpy as np
5 from numpy.testing import TestCase
6 from numpy.testing.utils import assert_equal, assert_almost_equal
7 from aubio import window, level_lin, db_spl, silence_detection, level_detection
8 from aubio import fvec, float_type
9
10 class 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
29 class 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
43 class 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
57 class 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
72 class 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
87 if __name__ == '__main__':
88     main()