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()