Changeset b7f3aaf


Ignore:
Timestamp:
Dec 5, 2009, 1:45:40 AM (15 years ago)
Author:
Paul Brossier <piem@piem.org>
Branches:
feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master, pitchshift, sampler, timestretch, yinfft+
Children:
f658288
Parents:
0f045b2
Message:

interfaces/python: start updating tests for mono

Location:
interfaces/python
Files:
1 added
1 edited

Legend:

Unmodified
Added
Removed
  • interfaces/python/test_aubio.py

    r0f045b2 rb7f3aaf  
    33from _aubio import *
    44from numpy import array
    5 
    6 AUBIO_DO_CASTING = 0
    75
    86class aubiomodule_test_case(TestCase):
     
    119    """ try importing aubio """
    1210
    13   def test_vector(self):
    14     a = fvec()
    15     a.length, a.channels
    16     a[0]
    17     array(a)
    18     a = fvec(10)
    19     a = fvec(1, 2)
    20     array(a).T
    21     a[0] = range(a.length)
    22     a[1][0] = 2
    23 
    24   def test_wrong_values(self):
    25     self.assertRaises (ValueError, fvec, -10)
    26     self.assertRaises (ValueError, fvec, 1, -1)
    27  
    28     a = fvec(2, 3)
    29     self.assertRaises (IndexError, a.__getitem__, 3)
    30     self.assertRaises (IndexError, a[0].__getitem__, 2)
    31 
    32   def test_alpha_norm_of_fvec(self):
    33     a = fvec(2, 2)
    34     self.assertEquals (alpha_norm(a, 1), 0)
    35     a[0] = [1, 2]
    36     self.assertEquals (alpha_norm(a, 1), 1.5)
    37     a[1] = [1, 2]
    38     self.assertEquals (alpha_norm(a, 1), 3)
    39     a[0] = [0, 1]; a[1] = [1, 0]
    40     self.assertEquals (alpha_norm(a, 2), 1)
    41 
    42   def test_alpha_norm_of_array_of_float32(self):
    43     # check scalar fails
    44     a = array(1, dtype = 'float32')
    45     self.assertRaises (ValueError, alpha_norm, a, 1)
    46     # check 3d array fails
    47     a = array([[[1,2],[3,4]]], dtype = 'float32')
    48     self.assertRaises (ValueError, alpha_norm, a, 1)
    49     # check 1d array
    50     a = array(range(10), dtype = 'float32')
    51     self.assertEquals (alpha_norm(a, 1), 4.5)
    52     # check 2d array
    53     a = array([range(10), range(10)], dtype = 'float32')
    54     self.assertEquals (alpha_norm(a, 1), 9)
    55 
    56   def test_alpha_norm_of_array_of_int(self):
    57     a = array(1, dtype = 'int')
    58     self.assertRaises (ValueError, alpha_norm, a, 1)
    59     a = array([[[1,2],[3,4]]], dtype = 'int')
    60     self.assertRaises (ValueError, alpha_norm, a, 1)
    61     a = array(range(10), dtype = 'int')
    62     self.assertRaises (ValueError, alpha_norm, a, 1)
    63 
    64   def test_alpha_norm_of_array_of_string (self):
    65     a = "hello"
    66     self.assertRaises (ValueError, alpha_norm, a, 1)
    67 
    68   def test_zero_crossing_rate(self):
    69     a = array([0,1,-1], dtype='float32')
    70     self.assertEquals (zero_crossing_rate(a), 1./3 )
    71     a = array([0.]*100, dtype='float32')
    72     self.assertEquals (zero_crossing_rate(a), 0 )
    73     a = array([-1.]*100, dtype='float32')
    74     self.assertEquals (zero_crossing_rate(a), 0 )
    75     a = array([1.]*100, dtype='float32')
    76     self.assertEquals (zero_crossing_rate(a), 0 )
    77 
    78   def test_alpha_norm_of_array_of_float64(self):
    79     # check scalar fail
    80     a = array(1, dtype = 'float64')
    81     self.assertRaises (ValueError, alpha_norm, a, 1)
    82     # check 3d array fail
    83     a = array([[[1,2],[3,4]]], dtype = 'float64')
    84     self.assertRaises (ValueError, alpha_norm, a, 1)
    85     if AUBIO_DO_CASTING:
    86       # check float64 1d array fail
    87       a = array(range(10), dtype = 'float64')
    88       self.assertEquals (alpha_norm(a, 1), 4.5)
    89       # check float64 2d array fail
    90       a = array([range(10), range(10)], dtype = 'float64')
    91       self.assertEquals (alpha_norm(a, 1), 9)
    92     else:
    93       # check float64 1d array fail
    94       a = array(range(10), dtype = 'float64')
    95       self.assertRaises (ValueError, alpha_norm, a, 1)
    96       # check float64 2d array fail
    97       a = array([range(10), range(10)], dtype = 'float64')
    98       self.assertRaises (ValueError, alpha_norm, a, 1)
    99 
    100   def test_fvec_min_removal_of_array(self):
    101     a = array([20,1,19], dtype='float32')
    102     b = min_removal(a)
    103     assert_equal (array(b), [19, 0, 18])
    104     assert_equal (b, [19, 0, 18])
    105     assert_equal (a, b)
    106     a[0] = 0
    107     assert_equal (a, b)
    108 
    109   def test_fvec_min_removal_of_array_float64(self):
    110     a = array([20,1,19], dtype='float64')
    111     if AUBIO_DO_CASTING:
    112       b = min_removal(a)
    113       assert_equal (array(b), [19, 0, 18])
    114       assert_equal (b, [19, 0, 18])
    115       #assert_equal (a, b)
    116     else:
    117       self.assertRaises (ValueError, min_removal, a)
    118      
    119   def test_fvec_min_removal_of_fvec(self):
    120     a = fvec(3, 1)
    121     a[0] = [20, 1, 19]
    122     b = min_removal(a)
    123     assert_equal (array(b), [19, 0, 18])
    124     assert_equal (b, [19, 0, 18])
    125     assert_equal (a, b)
    12611
    12712if __name__ == '__main__':
Note: See TracChangeset for help on using the changeset viewer.