make sure AtLeast collected at least N outputs even if reaching the end of input

1 files changed, 5 insertions, 1 deletions

diff --git a/src/lib.rs b/src/lib.rs
index d2af3ac..d96eb4c 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -338,7 +338,11 @@ where
             i += 1;
         }
 
-        Ok((it, outputs))
+        if i >= self.count {
+            Ok((it, outputs))
+        } else {
+            Err(EM::bind(|| it))
+        }
     }
 }