Update test_iter.py

Author: zhihao11ui

tests/test_iter.py

@@ -2,14 +2,6 @@
 from pydu.iter import first, last, all, any, join
 
 
-@pytest.mark.parametrize(
-    'iterable', (
-        [1, 2],
-        (1, 2),
-        {1, 2},
-        {1: 1, 2: 2},
-        iter([1, 2])
-    ))
 def first(iterable):
     return next(iter(iterable)) if hasattr(iterable, '__iter__') else iterable
 
@@ -18,6 +10,14 @@ def last(iterable):
     return list(iterable)[-1]
 
 
+@pytest.mark.parametrize(
+    'iterable', (
+        [1, 2],
+        (1, 2),
+        {1, 2},
+        {1: 1, 2: 2},
+        iter([1, 2])  # This is the iterator causing exhaustion
+    ))
 def test_first_last(iterable):
     if isinstance(iterable, (list, tuple, set, dict)):
         assert first(iterable) == 1