#! /usr/bin/env python
import numpy as np
-from numpy.testing import TestCase, assert_equal, assert_almost_equal
+from numpy.testing import TestCase, assert_equal
from aubio import cvec, fvec, float_type
wrong_type = 'float32' if float_type == 'float64' else 'float64'
assert_equal(spec.phas, 0)
def test_assign_cvec_phas_slice(self):
+ if float_type == 'float64':
+ pi = np.float64(np.pi)
+ else:
+ pi = np.float32(np.pi)
spec = cvec(1024)
- spec.phas[39:-1] = -np.pi
+ spec.phas[39:-1] = -pi
assert_equal(spec.phas[0:39], 0)
- assert_almost_equal(spec.phas[39:-1], -np.pi, decimal=7)
+ assert_equal(spec.phas[39:-1], -pi)
assert_equal(spec.norm, 0)
def test_assign_cvec_with_other_cvec(self):