From 71028fa45e7b65de62402d6a049afa16baca39e8 Mon Sep 17 00:00:00 2001
From: Lars-Dominik Braun <lars@6xq.net>
Date: Wed, 25 Nov 2020 11:13:20 +0100
Subject: Add a few more tests

---
 lulua/util.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'lulua/util.py')

diff --git a/lulua/util.py b/lulua/util.py
index c5634c2..5d7ea1b 100644
--- a/lulua/util.py
+++ b/lulua/util.py
@@ -30,7 +30,10 @@ def limit (l, n):
     """ Limit the number of items drawn from iterable l to n. """
     it = iter (l)
     for i in range (n):
-        yield next (it)
+        try:
+            yield next (it)
+        except StopIteration:
+            break
 
 class YamlLoader:
     """
-- 
cgit v1.2.3