iterators

import time from iterators import TimeoutIterator def my_slow_iterator(): yield 1 time.sleep(0.6) # This sleep might cause a timeout yield 2 time.sleep(0.4) yield 3 print("--- Demonstrating TimeoutIterator ---") # Example 1: Basic iteration i = my_slow_iterator() it = TimeoutIterator(i) print(f"Next: {next(it)}") # Expected: 1 print(f"Next: {next(it)}") # Expected: 2 (if timeout not hit) print(f"Next: {next(it)}") # Expected: 3 try: next(it) except StopIteration: print("StopIteration: Iterator exhausted.") print("\n--- Demonstrating Timeout with Sentinel ---") i_timeout = my_slow_iterator() it_timeout = TimeoutIterator(i_timeout, timeout=0.5) print(f"Next (with timeout): {next(it_timeout)}") # Expected: 1 # The next call will likely hit the timeout if sleep is 0.6s and timeout 0.5s sentinel_value = next(it_timeout) if sentinel_value == it_timeout.get_sentinel(): print("Timeout occurred, sentinel value returned.") else: print(f"Next (after potential timeout): {sentinel_value}") # Continue to consume remaining items, potentially with adjusted timeout it_timeout.set_timeout(0.7) # Adjust timeout dynamically try: print(f"Next (after timeout reset): {next(it_timeout)}") print(f"Next (final item): {next(it_timeout)}") except StopIteration: print("StopIteration: Iterator exhausted after timeout handling.")