Fix unittests for 'sets' in python3.

zopefoundation · Jun 8, 2017 · 8c55dd9 · 8c55dd9
1 parent 42905e1
commit 8c55dd9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/Products/PythonScripts/tests/testPythonScript.py b/src/Products/PythonScripts/tests/testPythonScript.py
@@ -11,6 +11,7 @@
 #
 ##############################################################################
 import os
+import six
 import unittest
 import warnings
 
@@ -224,7 +225,12 @@ def testGetSize(self):
         f = self._filePS('complex_print')
         self.assertEqual(f.get_size(), len(f.read()))
 
-    def testSet(self):
+    def testBuiltinSet(self):
+        res = self._newPS('return len(set([1, 2, 3, 1]))')()
+        self.assertEqual(res, 3)
+
+    @unittest.skipIf(six.PY3, 'sets module does not exist in python3')
+    def testSetModule(self):
         res = self._newPS('from sets import Set; return len(Set([1,2,3]))')()
         self.assertEqual(res, 3)