fix contagion in IOTA when step is a complex integer and start is a float

diff --git a/numbers.lisp b/numbers.lisp
index 9ced84b..beebb58 100644 (file)
--- a/numbers.lisp
+++ b/numbers.lisp
@@ -58,9 +58,9 @@ Examples:
 "
   (declare (type (integer 0) n) (number start step))
   (loop repeat n
-     ;; KLUDGE: get numeric contagion right for the first element too
-     for i = (+ start (- step step)) then (+ i step)
-     collect i))
+        ;; KLUDGE: get numeric contagion right for the first element too
+        for i = (+ (- (+ start step) step)) then (+ i step)
+        collect i))
 
 (declaim (inline map-iota))
 (defun map-iota (function n &key (start 0) (step 1))

diff --git a/tests.lisp b/tests.lisp
index 198104c..f203aad 100644 (file)
--- a/tests.lisp
+++ b/tests.lisp
@@ -1880,3 +1880,8 @@
   t
   t
   t)
+
+(deftest iota.fp-start-and-complex-integer-step
+    (equal '(#C(0.0 0.0) #C(0.0 2.0) #C(0.0 4.0))
+           (iota 3 :start 0.0 :step #C(0 2)))
+  t)