[latebinding] relaz requirements from generator to iterator

Author: apalala

Diff for: Makefile

@@ -1,4 +1,8 @@
-test: pytest mypy
+test: lint pytest mypy
+
+
+lint:
+	ruff late test
 
 pytest:
 	pytest -v

Diff for: README.md

@@ -116,9 +116,9 @@ assert d.x is not c.x
 ```
 
 
-### Working with generators
+### Working with iterators
 
-**Late** allows passing a generator as a default argument value, 
+**Late** allows passing an iterator as a default argument value, 
 and it will provide the next value on each function call. The usefulness of
 this feature is unknown, but it's something that came up during the discussions
 about default arguments, so **Late** implements it.
@@ -144,7 +144,7 @@ about default arguments, so **Late** implements it.
     assert f() == 5
 ```
 
-This is a possible use for the generators feature. Imagine a function that requires a unique ID, and 
+This is a possible use for the iterator feature. Imagine a function that requires a unique ID, and 
 will generate one if none is provided. Without **Late** the declaration would be:
 
 ```python
@@ -191,7 +191,7 @@ For values of immutable types, ``__()`` will return the same value. For all othe
 will wrap the value in a special ``namedtuple(actual=value)``. At function invocation time, this it what happens:
 
 * if the argument name is already in ``kwargs``, nothing is done
-* if the wrapped value is a generator, then ``next(actual)`` is used
+* if the wrapped value is an iterator, then ``next(actual)`` is used
 * if the wrapped value is a function, then ``actual()`` is used
 * in all other cases ``copy.deepcopy(actual)`` is used
 

Diff for: late/__init__.py

@@ -32,7 +32,7 @@ def late(o: _T | Iterator[_V] | Callable[[], _R]) -> _T | _V | _R:
 def _lateargs(func: Callable, **kwargs) -> dict[str, Any]:
 
     def resolve_default(value):
-        if inspect.isgenerator(value):
+        if isinstance(value, Iterator):
             return next(value)
         if inspect.isfunction(value):
             return value()

Diff for: test/complex_test.py

@@ -1,7 +1,6 @@
-import inspect
 from typing import Any
 
-from late import latebinding, __, _LateBound
+from late import latebinding, __
 
 
 def test_recursion():

Diff for: test/generator_test.py

@@ -1,5 +1,5 @@
 from typing import Iterator
-from late import latebinding, __, _LateBound
+from late import latebinding, __
 
 
 def test_generator():

Diff for: test/iterator_test.py

@@ -0,0 +1,13 @@
+from itertools import count
+
+from late import latebinding, __
+
+
+def test_iterator():
+
+    @latebinding
+    def f(x: int = __(count(start=7))):
+        return x
+
+    assert f() == 7
+    assert f() == 8