annotate test/test_collect.py @ 88:7228921e8425

Some way to implementing collect()

import vamp
import numpy as np

plugin_key = "vamp-test-plugin:vamp-test-plugin"
plugin_key_freq = "vamp-test-plugin:vamp-test-plugin-freq"

rate = 44100

# Throughout this file we have the assumption that the plugin gets run with a
# blocksize of 1024, and with a step of 1024 for the time-domain version or 512
# 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

def input_data(n):
    # start at 1, not 0 so that all elts are non-zero
    return np.arange(n) + 1    

def test_collect_runs_at_all():
    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)