shithub: aubio

Info  •  Files  •  Log  •  Branches

tests/python: update template.py

--- a/tests/python/template.py

+++ b/tests/python/template.py

@@ -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 @@

       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)