From 95e3cba8835a77e9042793b4bfcb71f44384e7f1 Mon Sep 17 00:00:00 2001 From: Paul Brossier Date: Sat, 15 Sep 2018 16:20:21 +0200 Subject: [PATCH] python/tests/test_dct.py: use float_type --- python/tests/test_dct.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/tests/test_dct.py b/python/tests/test_dct.py index 07c336d9..fbcbde87 100755 --- a/python/tests/test_dct.py +++ b/python/tests/test_dct.py @@ -24,19 +24,19 @@ class aubio_dct(TestCase): """ test that dct(arange(8)) is computed correctly >>> from scipy.fftpack import dct - >>> a_in = np.arange(8).astype('float32') + >>> a_in = np.arange(8).astype(aubio.float_type) >>> precomputed = dct(a_in, norm='ortho') """ N = len(precomputed_arange) a_dct = aubio.dct(8) - a_in = np.arange(8).astype('float32') + a_in = np.arange(8).astype(aubio.float_type) a_expected = aubio.fvec(precomputed_arange) assert_almost_equal(a_dct(a_in), a_expected, decimal=6) def test_some_ones(self): """ test that dct(somevector) is computed correctly """ a_dct = aubio.dct(16) - a_in = np.ones(16).astype('float32') + a_in = np.ones(16).astype(aubio.float_type) a_in[1] = 0 a_in[3] = np.pi a_expected = aubio.fvec(precomputed_some_ones) @@ -45,7 +45,7 @@ class aubio_dct(TestCase): def test_reconstruction(self): """ test that some_ones vector can be recontructed """ a_dct = aubio.dct(16) - a_in = np.ones(16).astype('float32') + a_in = np.ones(16).astype(aubio.float_type) a_in[1] = 0 a_in[3] = np.pi a_dct_in = a_dct(a_in) -- 2.11.0