[py] improve cvec test using assert_equal and aubio float_type (closes gh-405)
authorPaul Brossier <piem@piem.org>
Tue, 18 Nov 2025 08:39:41 +0000 (09:39 +0100)
committerPaul Brossier <piem@piem.org>
Tue, 18 Nov 2025 08:39:41 +0000 (09:39 +0100)
python/tests/test_cvec.py

index 1b7b1ec..90876bf 100755 (executable)
@@ -1,7 +1,7 @@
 #! /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'
@@ -40,10 +40,14 @@ class aubio_cvec_test_case(TestCase):
         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):