From 9f8cd9f3633dab32802af562fc55959decf4ead6 Mon Sep 17 00:00:00 2001 From: Paul Brossier Date: Mon, 2 May 2016 12:46:30 +0200 Subject: [PATCH] python/tests/test_cvec.py: simplify, use nose2 --- python/tests/test_cvec.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/python/tests/test_cvec.py b/python/tests/test_cvec.py index 1f3c9eca..00751c7a 100755 --- a/python/tests/test_cvec.py +++ b/python/tests/test_cvec.py @@ -1,16 +1,16 @@ #! /usr/bin/env python -from numpy.testing import TestCase, run_module_suite +from numpy.testing import TestCase from numpy.testing import assert_equal, assert_almost_equal -from aubio import cvec -from numpy import array, shape, pi +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) - shape(a.norm) - shape(a.phas) + 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.) @@ -41,11 +41,11 @@ class aubio_cvec_test_case(TestCase): def test_assign_cvec_phas_slice(self): spec = cvec(1024) - spec.phas[39:-1] = -pi + spec.phas[39:-1] = -np.pi assert_equal(spec.phas[0:39], 0) - assert_equal(spec.phas[39:-1], -pi) + assert_equal(spec.phas[39:-1], -np.pi) assert_equal(spec.norm, 0) if __name__ == '__main__': - from unittest import main + from nose2 import main main() -- 2.11.0