comparison test/test_collect.py @ 92:18b412a9c4d5

Tests for timestamp filling

 # for the frequency-domain one. That is certainly expected to be the norm for a
 # plugin like this that declares no preference, and the Python Vamp module is
 # expected to follow the norm
 
 blocksize = 1024
+eps = 1e-6
 
 def input_data(n):
     # start at 1, not 0 so that all elts are non-zero
     return np.arange(n) + 1    
 

     results = list(vamp.collect(buf, rate, plugin_key, "input-timestamp"))
     assert len(results) == 10
     for r in results:
         assert r["timestamp"] == vamp.vampyhost.frame_to_realtime(r["values"][0], rate)
 
+def test_collect_fixed_sample_rate():
+    buf = input_data(blocksize * 10)
+    results = list(vamp.collect(buf, rate, plugin_key, "curve-fsr"))
+    assert len(results) == 10
+    i = 0
+    for r in results:
+        assert r["timestamp"] == vamp.vampyhost.RealTime('seconds', i * 0.4)
+        assert abs(r["values"][0] - i * 0.1) < eps
+        i = i + 1
+
+def test_collect_fixed_sample_rate_2():
+    buf = input_data(blocksize * 10)
+    results = list(vamp.collect(buf, rate, plugin_key, "curve-fsr-timed"))
+    assert len(results) == 10
+    i = 0
+    for r in results:
+        assert r["timestamp"] == vamp.vampyhost.RealTime('seconds', i * 0.4)
+        assert abs(r["values"][0] - i * 0.1) < eps
+        i = i + 1
+        
+def test_collect_variable_sample_rate():
+    buf = input_data(blocksize * 10)
+    results = list(vamp.collect(buf, rate, plugin_key, "curve-vsr"))
+    assert len(results) == 10
+    i = 0
+    for r in results:
+        assert r["timestamp"] == vamp.vampyhost.RealTime('seconds', i * 0.75)
+        assert abs(r["values"][0] - i * 0.1) < eps
+        i = i + 1