ref: 9f8cd9f3633dab32802af562fc55959decf4ead6
dir: /python/tests/test_cvec.py/
#! /usr/bin/env python
from numpy.testing import TestCase
from numpy.testing import assert_equal, assert_almost_equal
from aubio import cvec, float_type
import numpy as np
class aubio_cvec_test_case(TestCase):
def test_vector_created_with_zeroes(self):
a = cvec(10)
assert_equal(a.norm.shape[0], 10 / 2 + 1)
assert_equal(a.phas.shape[0], 10 / 2 + 1)
a.norm[0]
assert_equal(a.norm, 0.)
assert_equal(a.phas, 0.)
def test_vector_assign_element(self):
a = cvec()
a.norm[0] = 1
assert_equal(a.norm[0], 1)
a.phas[0] = 1
assert_equal(a.phas[0], 1)
def test_vector_assign_element_end(self):
a = cvec()
a.norm[-1] = 1
assert_equal(a.norm[-1], 1)
assert_equal(a.norm[len(a.norm)-1], 1)
a.phas[-1] = 1
assert_equal(a.phas[-1], 1)
assert_equal(a.phas[len(a.phas)-1], 1)
def test_assign_cvec_norm_slice(self):
spec = cvec(1024)
spec.norm[40:100] = 100
assert_equal(spec.norm[0:40], 0)
assert_equal(spec.norm[40:100], 100)
assert_equal(spec.norm[100:-1], 0)
assert_equal(spec.phas, 0)
def test_assign_cvec_phas_slice(self):
spec = cvec(1024)
spec.phas[39:-1] = -np.pi
assert_equal(spec.phas[0:39], 0)
assert_equal(spec.phas[39:-1], -np.pi)
assert_equal(spec.norm, 0)
if __name__ == '__main__':
from nose2 import main
main()