Merge pull request RustPython#940 from sanxiyn/test-all-any

Test all and any

Author: coolreader18

tests/snippets/builtin_all.py

@@ -0,0 +1,10 @@
+from testutils import assert_raises
+from testutils import TestFailingBool, TestFailingIter
+
+assert all([True])
+assert not all([False])
+assert all([])
+assert not all([False, TestFailingBool()])
+
+assert_raises(RuntimeError, lambda: all(TestFailingIter()))
+assert_raises(RuntimeError, lambda: all([TestFailingBool()]))

tests/snippets/builtin_any.py

@@ -1,13 +1,10 @@
-assert any([1]);
-assert not any([]);
-assert not any([0,0,0,0]);
-assert any([0,0,1,0,0]);
-def anything(a):
-    return a
+from testutils import assert_raises
+from testutils import TestFailingBool, TestFailingIter
 
-class Test:
- def __iter__(self):
-   while True:
-    yield True
+assert any([True])
+assert not any([False])
+assert not any([])
+assert any([True, TestFailingBool()])
 
-assert any(map(anything, Test()))
+assert_raises(RuntimeError, lambda: any(TestFailingIter()))
+assert_raises(RuntimeError, lambda: any([TestFailingBool()]))

tests/snippets/testutils.py

@@ -34,3 +34,12 @@ def __exit__(self, exc_type, exc_val, exc_tb):
         if not issubclass(exc_type, self.expected):
             return False
         return True
+
+
+class TestFailingBool:
+    def __bool__(self):
+        raise RuntimeError
+
+class TestFailingIter:
+    def __iter__(self):
+        raise RuntimeError