use next() built-in to get first item of iterable

ActivisionGameScience · May 12, 2016 · 25f9ef6 · 25f9ef6
1 parent bb329cf
commit 25f9ef6
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/assertpy/assertpy.py b/assertpy/assertpy.py
@@ -485,10 +485,9 @@ def starts_with(self, prefix):
         elif isinstance(self.val, collections.Iterable):
             if len(self.val) == 0:
                 raise ValueError('val must not be empty')
-            for i in self.val:
-                if i != prefix:
-                    self._err('Expected %s to start with <%s>, but did not.' % (self.val, prefix))
-                break
+            first = next(i for i in self.val)
+            if first != prefix:
+                self._err('Expected %s to start with <%s>, but did not.' % (self.val, prefix))
         else:
             raise TypeError('val is not a string or iterable')
         return self