feature/autosinkfeature/constantqfeature/pitchshiftfeature/timestretch
Last change
on this file since 51b5f9c was
319edae,
checked in by Paul Brossier <piem@piem.org>, 13 months ago

[tests] avoid some imports, move main to end


Property mode set to
100644

File size:
1.1 KB

Line  

1  #! /usr/bin/env python 

2  

3  import numpy as np 

4  from numpy.testing import TestCase, assert_equal 

5  import aubio 

6  

7  class aubio_shift_test_case(TestCase): 

8  

9  def run_shift_ishift(self, n): 

10  ramp = np.arange(n, dtype=aubio.float_type) 

11  # construct expected output 

12  # even length: [5. 6. 7. 8. 9. 0. 1. 2. 3. 4.] 

13  # odd length: [4. 5. 6. 0. 1. 2. 3.] 

14  half = n  n//2 

15  expected = np.concatenate([np.arange(half, n), np.arange(half)]) 

16  # shift in place, returns modified copy 

17  assert_equal(aubio.shift(ramp), expected) 

18  # check input was changed as expected 

19  assert_equal(ramp, expected) 

20  # construct expected output 

21  expected = np.arange(n) 

22  # revert shift in place, returns modifed copy 

23  assert_equal(aubio.ishift(ramp), expected) 

24  # check input was shifted back 

25  assert_equal(ramp, expected) 

26  

27  def test_can_shift_fvec(self): 

28  self.run_shift_ishift(10) 

29  

30  def test_can_shift_fvec_odd(self): 

31  self.run_shift_ishift(7) 

32  

33  if __name__ == '__main__': 

34  from unittest import main 

35  main() 

Note: See
TracBrowser
for help on using the repository browser.