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 ]