#! /usr/bin/env python from numpy.testing import TestCase, assert_equalfrom numpy import array, arange, isnan, isinffrom aubio import bintomidi, miditobin, freqtobin, bintofreq, freqtomidi, miditofreqfrom aubio import unwrap2pifrom aubio import fvecfrom math import pi class aubio_mathutils(TestCase):     def test_unwrap2pi(self):        unwrap2pi(int(23))        unwrap2pi(float(23.))        unwrap2pi(long(23.))        unwrap2pi(arange(10))        unwrap2pi(arange(10).astype("int"))        unwrap2pi(arange(10).astype("float"))        unwrap2pi(arange(10).astype("float32"))        unwrap2pi([1,3,5])        unwrap2pi([23.,24.,25.])        a = fvec(10)        a[:] = 4.        unwrap2pi(a)        a = pi/100. * arange(-600,600).astype("float")        b = unwrap2pi (a)        #print zip(a, b)         try:            print unwrap2pi(["23.","24.",25.])        except Exception, e:            pass     def test_unwrap2pi_takes_fvec(self):        a = fvec(10)        b = unwrap2pi(a)        #print zip(a, b)        assert ( b > -pi ).all()        assert ( b <= pi ).all()     def test_unwrap2pi_takes_array_of_float(self):        a = arange(-10., 10.).astype("float")        b = unwrap2pi(a)        #print zip(a, b)        assert ( b > -pi ).all()        assert ( b <= pi ).all()     def test_unwrap2pi_takes_array_of_float32(self):        a = arange(-10, 10).astype("float32")        b = unwrap2pi(a)        #print zip(a, b)        assert ( b > -pi ).all()        assert ( b <= pi ).all()     def test_freqtomidi(self):        a = array(range(-20, 50000, 100) + [ -1e32, 1e32 ])        b = freqtomidi(a)        #print zip(a, b)        assert_equal ( isnan(array(b)), False )        assert_equal ( isinf(array(b)), False )        assert_equal ( array(b) < 0, False )     def test_miditofreq(self):        a = range(-30, 200) + [-100000, 10000]        b = miditofreq(a)        #print zip(a, b)        assert_equal ( isnan(b), False )        assert_equal ( isinf(b), False )        assert_equal ( b < 0, False )     def test_miditobin(self):        a = range(-30, 200) + [-100000, 10000]        b = [ bintomidi(x, 44100, 512) for x in a ]        #print zip(a, b)        assert_equal ( isnan(array(b)), False )        assert_equal ( isinf(array(b)), False )        assert_equal ( array(b) < 0, False )     def test_bintomidi(self):        a = range(-100, 512)        b = [ bintomidi(x, 44100, 512) for x in a ]        #print zip(a, b)        assert_equal ( isnan(array(b)), False )        assert_equal ( isinf(array(b)), False )        assert_equal ( array(b) < 0, False )     def test_freqtobin(self):        a = range(-20, 50000, 100) + [ -1e32, 1e32 ]        b = [ freqtobin(x, 44100, 512) for x in a ]        #print zip(a, b)        assert_equal ( isnan(array(b)), False )        assert_equal ( isinf(array(b)), False )        assert_equal ( array(b) < 0, False )     def test_bintofreq(self):        a = range(-20, 148)        b = [ bintofreq(x, 44100, 512) for x in a ]        #print zip(a, b)        assert_equal ( isnan(array(b)), False )        assert_equal ( isinf(array(b)), False )        assert_equal ( array(b) < 0, False ) if __name__ == '__main__':    from unittest import main    main()