shithub: aubio

Download patch

ref: 203c551c7a66307c45e1887a2d3194893b2feecd
parent: b29565c132fa44d1e8de4f2b246e25b2383587dd
author: Paul Brossier <piem@piem.org>
date: Sun Sep 16 23:01:29 EDT 2007

added simple tests for fvec and cvec

--- a/python/test/all_tests.py
+++ b/python/test/all_tests.py
@@ -8,7 +8,7 @@
 
 import unittest
 
-modules_to_test = ['aubiomodule']
+modules_to_test = ['aubiomodule', 'fvec', 'cvec']
 
 if __name__ == '__main__':
   for module in modules_to_test: exec('from %s import *' % module)
--- /dev/null
+++ b/python/test/cvec.py
@@ -1,0 +1,51 @@
+import unittest
+
+from aubio.aubiowrapper import *
+
+buf_size = 2048
+channels = 3
+
+class cvec_test_case(unittest.TestCase):
+
+  def setUp(self):
+    self.vector = new_cvec(buf_size, channels)
+
+  def tearDown(self):
+    del_cvec(self.vector)
+
+  def test_cvec(self):
+    """ create and delete cvec """
+    pass
+
+  def test_cvec_read_norm(self):
+    """ check new cvec norm elements are set to 0. """
+    for index in range(buf_size/2+1):
+      for channel in range(channels):
+        self.assertEqual(cvec_read_norm(self.vector,channel,index),0.)
+
+  def test_cvec_read_phas(self):
+    """ check new cvec phas elements are set to 0. """
+    for index in range(buf_size/2+1):
+      for channel in range(channels):
+        self.assertEqual(cvec_read_phas(self.vector,channel,index),0.)
+
+  def test_cvec_write_norm(self):
+    """ check new cvec norm elements are set with cvec_write_norm """
+    for index in range(buf_size/2+1):
+      for channel in range(channels):
+        cvec_write_norm(self.vector,1.,channel,index)
+    for index in range(buf_size/2+1):
+      for channel in range(channels):
+        self.assertEqual(cvec_read_norm(self.vector,channel,index),1.)
+
+  def test_cvec_write_phas(self):
+    """ check new cvec phas elements are set with cvec_write_phas """
+    for index in range(buf_size/2+1):
+      for channel in range(channels):
+        cvec_write_phas(self.vector,1.,channel,index)
+    for index in range(buf_size/2+1):
+      for channel in range(channels):
+        self.assertEqual(cvec_read_phas(self.vector,channel,index),1.)
+
+if __name__ == '__main__':
+  unittest.main()
--- /dev/null
+++ b/python/test/fvec.py
@@ -1,0 +1,36 @@
+import unittest
+
+from aubio.aubiowrapper import *
+
+buf_size = 2048
+channels = 3
+
+class fvec_test_case(unittest.TestCase):
+
+  def setUp(self):
+    self.vector = new_fvec(buf_size, channels)
+
+  def tearDown(self):
+    del_fvec(self.vector)
+
+  def test_fvec(self):
+    """ create and delete fvec """
+    pass
+
+  def test_fvec_read_sample(self):
+    """ check new fvec elements are set to 0. """
+    for index in range(buf_size):
+      for channel in range(channels):
+        self.assertEqual(fvec_read_sample(self.vector,channel,index),0.)
+
+  def test_fvec_write_sample(self):
+    """ check new fvec elements are set with fvec_write_sample """
+    for index in range(buf_size):
+      for channel in range(channels):
+        fvec_write_sample(self.vector,1.,channel,index)
+    for index in range(buf_size):
+      for channel in range(channels):
+        self.assertEqual(fvec_read_sample(self.vector,channel,index),1.)
+
+if __name__ == '__main__':
+  unittest.main()