Mercurial > hg > vampy-host
diff test/test_collect.py @ 97:06c4afba4fc5
Fix matrix return
author | Chris Cannam |
---|---|
date | Tue, 03 Feb 2015 10:29:21 +0000 |
parents | f0e005248b9a |
children | 0c5a4dc04ed9 |
line wrap: on
line diff
--- a/test/test_collect.py Mon Feb 02 17:31:30 2015 +0000 +++ b/test/test_collect.py Tue Feb 03 10:29:21 2015 +0000 @@ -71,6 +71,5 @@ assert abs(float(step) - (1024.0 / rate)) < eps assert len(results) == 10 for i in range(len(results)): - expected = [] - actual = results[i] - assert actual == expected + expected = np.array([ (j + i + 2.0) / 30.0 for j in range(0, 10) ]) + assert (abs(results[i] - expected) < eps).all()