diff test/test_collect.py @ 88:7228921e8425

Some way to implementing collect()

--- a/test/test_collect.py	Wed Jan 21 15:04:05 2015 +0000
+++ b/test/test_collect.py	Wed Jan 21 15:13:00 2015 +0000
@@ -20,6 +20,14 @@
     return np.arange(n) + 1    
 
 def test_collect_runs_at_all():
-    buf = input_data(blocksize)
-    results = vamp.collect(buf, rate, plugin_key, "input-summary")
-    assert type(results) == dict
+    buf = input_data(blocksize * 10)
+    results = list(vamp.collect(buf, rate, plugin_key, "input-timestamp"))
+    assert results != []
+
+def test_collect_one_sample_per_step():
+    buf = input_data(blocksize * 10)
+    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)
+