changeset 44:af167d5895a3

More channel tests

--- a/test_process.py	Wed Dec 03 08:25:05 2014 +0000
+++ b/test_process.py	Wed Dec 03 10:33:58 2014 +0000
@@ -70,4 +70,41 @@
     result = plug.process([[3,3]], vh.RealTime(0, 0))
     assert result[8] == [ { "label" : "", "values" : np.array([3.0]) } ]
 
+def test_process_output_2ch():
+    plug = vh.loadPlugin(testPluginKey, rate)
+    plug.initialise(2, 2, 2)
+    try:
+        # Too few channels
+        result = plug.process([[3,4]], vh.RealTime(0, 0))
+        assert False
+    except TypeError:
+        pass
+    try:
+        # Too many channels
+        result = plug.process([[3,4],[5,6],[7,8]], vh.RealTime(0, 0))
+        assert False
+    except TypeError:
+        pass
+    result = plug.process([[3,3],[4,4]], vh.RealTime(0, 0))
+    assert (result[8][0]["values"] == np.array([3.0,4.0])).all()
+
+def test_process_output_3ch():
+    plug = vh.loadPlugin(testPluginKey, rate)
+    plug.initialise(3, 2, 2)
+    try:
+        # Too few channels
+        result = plug.process([[3,4],[5,6]], vh.RealTime(0, 0))
+        assert False
+    except TypeError:
+        pass
+    try:
+        # Too many channels
+        result = plug.process([[3,4],[5,6],[7,8],[9,10]], vh.RealTime(0, 0))
+        assert False
+    except TypeError:
+        pass
+    result = plug.process([[3,3],[4,4],[5,5]], vh.RealTime(0, 0))
+    assert (result[8][0]["values"] == np.array([3.0,4.0,5.0])).all()
+
+