bug fix: arange (when step < 0)

Previous arange function have bug when called like arange(10, 1, -1)
GBR-613 · Nov 4, 2016 · 299b51b · 299b51b
1 parent e2fec3c
commit 299b51b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/smop/core.py b/smop/core.py
@@ -361,8 +361,9 @@ def arange(start,stop,step=1,**kwargs):
     >>> size(a)
     matlabarray([[ 1, 10]])
     """
+    expand_value = 1 if step > 0 else -1
     return matlabarray(np.arange(start,
-                                 stop+1,
+                                 stop+expand_value,
                                  step,
                                  **kwargs).reshape(1,-1),**kwargs)
 def cat(*args):