add 1 /to/ ...

Author: czheo

syntax_sugar/__init__.py

@@ -5,3 +5,4 @@
 from ._stream import *
 from ._placeholder import *
 from ._match import *
+from ._iter import *

syntax_sugar/_iter.py

@@ -1,8 +1,11 @@
 from ._pipe import pipe
 from itertools import tee, islice
 
+__all__ = [
+    'INF',
+]
+
 INF = float('inf')
-NEGINF = float('-inf')
 
 class Iterator:
     def __init__(self, data):
@@ -33,11 +36,14 @@ def slice(self, start=0, stop=None, step=1):
 
 class Range:
     def __init__(self, start, end):
-        if start in {INF, NEGINF}:
+        if start in {INF, -INF}:
             raise ValueError('Cannot start range from infinity')
 
+        if end == Ellipsis:
+            end = INF
+
         valid_char = lambda c: isinstance(c, str) and len(c) == 1
-        valid_integer = lambda i: isinstance(i, int) or i == INF or i == NEGINF
+        valid_integer = lambda i: isinstance(i, int) or i == INF or i == -INF
 
         if valid_integer(start) and valid_integer(end):
             self.type = 'number'

syntax_sugar/_util.py

@@ -7,14 +7,11 @@
     'as_a',
     'to',
     'step',
-    'INF',
     'has',
     'take',
     'drop',
 ]
 
-INF = float('inf')
-
 def flip(fn):
     def wrapper(*args):
         return fn(args[1], args[0])

tests/test_iter.py

@@ -1,6 +1,6 @@
 from pytest import raises
-from syntax_sugar._iter import Range, Iterator
-from syntax_sugar._util import take, to, INF
+from syntax_sugar._iter import Range, Iterator, INF
+from syntax_sugar._util import take, to
 from itertools import product