adding tests for set

zopefoundation · Feb 28, 2017 · 9842dc8 · 9842dc8
1 parent 092328a
commit 9842dc8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/RestrictedPython/tests/restricted_module.py b/src/RestrictedPython/tests/restricted_module.py
@@ -33,6 +33,11 @@ def try_map():
     print map(inc, x),
     return printed
 
+def try_set():
+    x = [1, 1, 2, 3]
+    print set(x)
+    return printed
+
 def try_apply():
     def f(x, y, z):
         return x + y + z

diff --git a/src/RestrictedPython/tests/testRestrictions.py b/src/RestrictedPython/tests/testRestrictions.py
@@ -251,6 +251,10 @@ def checkTryMap(self):
         res = self.execFunc('try_map')
         self.assertEqual(res, "[2, 3, 4]")
 
+    def checkTrySet(self):
+        res = self.execFunc('try_set')
+        self.assertEqual(res, "set([1, 2, 3])")
+
     def checkApply(self):
         del apply_wrapper_called[:]
         res = self.execFunc('try_apply')