35 lines · 1.1 KB · Python
RawDownload
#! /usr/bin/env python import numpy as npfrom numpy.testing import TestCase, assert_equalimport aubio class aubio_shift_test_case(TestCase):     def run_shift_ishift(self, n):        ramp = np.arange(n, dtype=aubio.float_type)        # construct expected output        # even length: [5. 6. 7. 8. 9. 0. 1. 2. 3. 4.]        # odd length: [4. 5. 6. 0. 1. 2. 3.]        half = n - n//2        expected = np.concatenate([np.arange(half, n), np.arange(half)])        # shift in place, returns modified copy        assert_equal(aubio.shift(ramp), expected)        # check input was changed as expected        assert_equal(ramp, expected)        # construct expected output        expected = np.arange(n)        # revert shift in place, returns modifed copy        assert_equal(aubio.ishift(ramp), expected)        # check input was shifted back        assert_equal(ramp, expected)     def test_can_shift_fvec(self):        self.run_shift_ishift(10)     def test_can_shift_fvec_odd(self):        self.run_shift_ishift(7) from unittest import mainif __name__ == '__main__':    main()

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t