Tests: add a stress test for analyzer. At first, test analyzers with nul and constant...

diff --git a/tests/test_analyzers_stress.py b/tests/test_analyzers_stress.py
new file mode 100644 (file)
index 0000000..453bbf2
--- /dev/null
+++ b/tests/test_analyzers_stress.py
@@ -0,0 +1,84 @@
+#! /usr/bin/env python
+
+# Author : Thomas Fillon <thomas@parisson.com>
+
+
+from unit_timeside import unittest, TestRunner
+import timeside
+import numpy as np
+
+
+class TestAnalyzers_with_zeros(unittest.TestCase):
+    """Stress test for analyzer with null input"""
+
+    def setUp(self):
+        import timeside.analyzer
+        samplerate = 16000  # LimsiSad require Fs = 16000 Hz
+        duration = 10
+        samples = np.zeros((duration * samplerate, 1))
+        self.decoder = timeside.decoder.ArrayDecoder(samples,
+                                                     samplerate=samplerate)
+
+    def _perform_test(self, analyzer_cls):
+        """Internal function that test if there is NaN in the results
+        of a given analyzer"""
+
+        pipe = (self.decoder | analyzer_cls())
+        pipe.run()
+        for key, result in pipe.results.items():
+            if 'value' in result.data_object.keys():
+                self.assertFalse(np.any(np.isnan(result.data)),
+                                 'NaN in %s data value' % result.name)
+
+
+class TestAnalyzers_withDC(TestAnalyzers_with_zeros):
+    """Stress test for analyzer with constant input"""
+
+    def setUp(self):
+        import timeside.analyzer
+        samplerate = 16000  # LimsiSad require Fs = 16000 Hz
+        duration = 10
+        samples = -1000*np.ones((duration * samplerate, 1))
+        self.decoder = timeside.decoder.ArrayDecoder(samples,
+                                                     samplerate=samplerate)
+
+
+def _tests_factory(test_class, test_doc, list_analyzers, skip_reasons={}):
+    """Define a test for each analyzer provided in the list"""
+    for analyzer in list_analyzers:
+
+        def test_func_factory(analyzer):
+            test_func = lambda self: self._perform_test(analyzer)
+            test_func.__doc__ = test_doc % analyzer.__name__
+            return test_func
+
+        test_func_name = "test_%s" % analyzer.__name__
+        test_func = test_func_factory(analyzer)
+
+        if analyzer.__name__ not in skip_reasons:
+            setattr(test_class, test_func_name,
+                    test_func_factory(analyzer))
+        else:  # Decorate with unittest.skip to skip test
+            setattr(test_class, test_func_name,
+                    unittest.skip(skip_reasons[analyzer.__name__])(test_func))
+
+
+# Define test to skip and corresponding reasons
+skip_reasons = {'VampSimpleHost': ('VampSimpleHost bypasses the decoder '
+                                   'and requires a file input')}
+
+# For each analyzer in TimeSide, define a "NaN test"
+_tests_factory(test_class=TestAnalyzers_withDC,
+               test_doc="test if %s returns NaN",
+               list_analyzers=timeside.core.processors(timeside.api.IAnalyzer),
+               skip_reasons=skip_reasons)
+
+# For each analyzer in TimeSide, define a "NaN test"
+_tests_factory(test_class=TestAnalyzers_with_zeros,
+               test_doc="test if %s returns NaN",
+               list_analyzers=timeside.core.processors(timeside.api.IAnalyzer),
+               skip_reasons=skip_reasons)
+
+
+if __name__ == '__main__':
+    unittest.main(testRunner=TestRunner())