Mercurial > hg > vampy-host
comparison test/test_collect.py @ 150:37d2fd57723e tracks
Restore backward compatibility for last change: return vector as well as tracks
author | Chris Cannam |
---|---|
date | Fri, 21 Apr 2017 14:22:42 +0100 |
parents | 65eeb604253f |
children |
comparison
equal
deleted
inserted
replaced
149:65eeb604253f | 150:37d2fd57723e |
---|---|
88 assert abs(float(step) - 0.4) < eps | 88 assert abs(float(step) - 0.4) < eps |
89 assert len(results) == 10 | 89 assert len(results) == 10 |
90 for i in range(len(results)): | 90 for i in range(len(results)): |
91 assert abs(results[i] - i * 0.1) < eps | 91 assert abs(results[i] - i * 0.1) < eps |
92 | 92 |
93 def test_collect_fixed_sample_rate_2(): | 93 def test_collect_fixed_sample_rate_2_vector(): |
94 # This one has discontinuities and overlaps, so it should be | |
95 # returned to us in two forms: as a simple vector, and as an | |
96 # additional tracks shape with one vector for each separate | |
97 # bit. This test covers the simple vector part | |
98 buf = input_data(blocksize * 10) | |
99 rdict = vamp.collect(buf, rate, plugin_key, "curve-fsr-timed") | |
100 step, results = rdict["vector"] | |
101 assert abs(float(step) - 0.4) < eps | |
102 assert len(results) == 10 | |
103 for i in range(len(results)): | |
104 assert abs(results[i] - i * 0.1) < eps | |
105 | |
106 def test_collect_fixed_sample_rate_2_tracks(): | |
107 # This one has discontinuities and overlaps, so it should be | |
108 # returned to us in two forms: as a simple vector, and as an | |
109 # additional tracks shape with one vector for each separate | |
110 # bit. This test covers the tracks part. | |
94 buf = input_data(blocksize * 10) | 111 buf = input_data(blocksize * 10) |
95 rdict = vamp.collect(buf, rate, plugin_key, "curve-fsr-timed") | 112 rdict = vamp.collect(buf, rate, plugin_key, "curve-fsr-timed") |
96 results = rdict["tracks"] | 113 results = rdict["tracks"] |
97 assert len(results) == 8 | 114 assert len(results) == 8 |
98 expected_starts = [ 0.0, 0.0, 0.0, 2.0, 2.0, 2.0, 4.0, 4.0 ] | 115 expected_starts = [ 0.0, 0.0, 0.0, 2.0, 2.0, 2.0, 4.0, 4.0 ] |