tests/python: update template.py
authorPaul Brossier <piem@piem.org>
Thu, 8 Oct 2009 00:10:10 +0000 (02:10 +0200)
committerPaul Brossier <piem@piem.org>
Thu, 8 Oct 2009 00:10:10 +0000 (02:10 +0200)
tests/python/template.py

index 05d1374ea03fbf58ef9a6660a6a43c6a212add2b..00e5344b9758011677712898dca7a7de3ca8dbdc 100644 (file)
@@ -1,6 +1,8 @@
 import unittest
+from numpy import array
 
 class aubio_unit_template(unittest.TestCase):
+  """ a class derivated from unittest.TestCase """
   
   def assertCloseEnough(self, first, second, places=5, msg=None):
     """Fail if the two objects are unequal as determined by their
@@ -15,3 +17,7 @@ class aubio_unit_template(unittest.TestCase):
       if round((second-first)/first, places) != 0:
         raise self.failureException, \
               (msg or '%r != %r within %r places' % (first, second, places))
+
+def array_from_text_file(filename, dtype = 'float'):
+  return array([line.split() for line in open(filename).readlines()], 
+      dtype = dtype)