Success import and better peaks validation in flatfilt and convfilt.
[hooke.git] / hooke / plugin / flatfilt.py
index f1e427e81d33914d86bd0a5d210b3c41937544e1..db54263bb9741ebe609ac12fbe8a480da05b5d6e 100644 (file)
@@ -38,7 +38,7 @@ from multiprocessing import Queue
 from numpy import diff
 from scipy.signal.signaltools import medfilt
 
-from ..command import Command, Argument, Failure
+from ..command import Command, Argument, Success, Failure
 from ..config import Setting
 from ..experiment import VelocityClamp
 from ..plugin import Plugin, argument_to_setting
@@ -202,7 +202,8 @@ class FlatFilterCommand (FilterCommand):
                         if c.name=='flat filter peaks'][0]
         conv_command.run(hooke, inq, outq, **params)
         peaks = outq.get()
-        if not isinstance(peaks[0], Peak):
+        if not (isinstance(peaks, list) and (len(peaks) == 0
+                                             or isinstance(peaks[0], Peak))):
             raise Failure('Expected a list of Peaks, not %s' % peaks)
         ret = outq.get()
         if not isinstance(ret, Success):