From 203c551c7a66307c45e1887a2d3194893b2feecd Mon Sep 17 00:00:00 2001
From: Paul Brossier <piem@piem.org>
Date: Mon, 17 Sep 2007 03:01:29 +0200
Subject: [PATCH] added simple tests for fvec and cvec

---
 python/test/all_tests.py |  2 +-
 python/test/cvec.py      | 51 ++++++++++++++++++++++++++++++++++++++++
 python/test/fvec.py      | 36 ++++++++++++++++++++++++++++
 3 files changed, 88 insertions(+), 1 deletion(-)
 create mode 100644 python/test/cvec.py
 create mode 100644 python/test/fvec.py

diff --git a/python/test/all_tests.py b/python/test/all_tests.py
index 4024f3f6..a066ca3a 100755
--- a/python/test/all_tests.py
+++ b/python/test/all_tests.py
@@ -8,7 +8,7 @@ sys.path.append(os.path.join('..','aubio','.libs'))
 
 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)
diff --git a/python/test/cvec.py b/python/test/cvec.py
new file mode 100644
index 00000000..f1043004
--- /dev/null
+++ b/python/test/cvec.py
@@ -0,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()
diff --git a/python/test/fvec.py b/python/test/fvec.py
new file mode 100644
index 00000000..5159af93
--- /dev/null
+++ b/python/test/fvec.py
@@ -0,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()
-- 
2.26.2