Test for enums + adjustments to Python-generation code.

1 files changed, 8 insertions, 8 deletions

diff --git a/stdlib/source/lux/abstract/enum.lux b/stdlib/source/lux/abstract/enum.lux
index 27690b286..9f074f7c5 100644
--- a/stdlib/source/lux/abstract/enum.lux
+++ b/stdlib/source/lux/abstract/enum.lux
@@ -9,13 +9,13 @@
   (: (-> e e) succ)
   (: (-> e e) pred))
 
-(def: (range' <= succ from to)
-  (All [a] (-> (-> a a Bit) (-> a a) a a (List a)))
-  (if (<= to from)
-    (#.Cons from (range' <= succ (succ from) to))
-    #.Nil))
-
-(def: #export (range (^open ",@.") from to)
+(def: #export (range enum from to)
   {#.doc "An inclusive [from, to] range of values."}
   (All [a] (-> (Enum a) a a (List a)))
-  (range' (order.<= ,@&order) ,@succ from to))
+  (let [(^open "/@.") enum
+        <= (order.<= /@&order)]
+    (loop [end to
+           output #.Nil]
+      (if (<= end from)
+        (recur (/@pred end) (#.Cons end output))
+        output))))