shithub: aubio

Info  •  Files  •  Log  •  Branches

python/lib/aubio/__init__.py: make sure length is not zero and float_type is imported

--- a/python/lib/aubio/__init__.py

+++ b/python/lib/aubio/__init__.py

@@ -2,6 +2,7 @@

 import numpy

 from ._aubio import *

+from ._aubio import float_type

 from .midiconv import *

 from .slicing import *

@@ -10,6 +11,8 @@

     def __new__(cls, input_arg=1024, **kwargs):

         if isinstance(input_arg, int):

+            if input_arg == 0:

+                raise ValueError("vector length of 1 or more expected")

             return numpy.zeros(input_arg, dtype=float_type, **kwargs)

         else:

             return numpy.array(input_arg, dtype=float_type, **kwargs)