Changes in tests/python/examples/aubioonset.py [100f950:f1771f9]

File:

• tests/python/examples/aubioonset.py (modified) (2 diffs)

tests/python/examples/aubioonset.py

@@ -1 +1 @@
 from template import *
 
-class aubioonset_test_case(program_test_case):
+class aubioonset_unit(program_test_case):
   
   import os.path
@@ -24 +24 @@
     self.getOutput()
     # only one onset in woodblock.aiff
-    assert len(str(self.output)) != 0, "no output produced with command:\n" \
-      + self.command
-    assert len(self.output.split('\n')) == 1
+    self.assertNotEqual(0, len(str(self.output)), \
+      "no output produced with command:\n" + self.command)
+    self.assertEqual(1, len(self.output.split('\n')) )
     # onset should be at 0.00000
-    assert float(self.output.strip()) == 0.
+    self.assertEqual(0, float(self.output.strip()))
 
-for name in ["energy", "specdiff", "hfc", "complex", "phase", "kl", "mkl"]:
-  exec("class aubioonset_test_case_"+name+"(aubioonset_test_case):\n\
+list_of_onset_modes = ["energy", "specdiff", "hfc", "complex", "phase", \
+                      "kl", "mkl", "specflux"]
+
+for name in list_of_onset_modes:
+  exec("class aubioonset_"+name+"_unit(aubioonset_unit):\n\
   options = \" -O "+name+" \"")