replace frozenset([...]) by frozenset((...))

--HG--
extra : amend_source : 7954c320c39d51cde9ac7d4368258ceb56a3bc2f

1 files changed, 9 insertions, 0 deletions

diff --git a/Cython/Compiler/Optimize.py b/Cython/Compiler/Optimize.py
index 88a857e6b..5e5b12767 100644
--- a/Cython/Compiler/Optimize.py
+++ b/Cython/Compiler/Optimize.py
@@ -1598,6 +1598,15 @@ class EarlyReplaceBuiltinCalls(Visitor.EnvTransform):
             return ExprNodes.AsTupleNode(node.pos, arg=result)
         return node
 
+    def _handle_simple_function_frozenset(self, node, pos_args):
+        """Replace frozenset([...]) by frozenset((...)) as tuples are more efficient.
+        """
+        if len(pos_args) != 1:
+            return node
+        if isinstance(pos_args[0], ExprNodes.ListNode):
+            pos_args[0] = pos_args[0].as_tuple()
+        return node
+
     def _handle_simple_function_list(self, node, pos_args):
         if not pos_args:
             return ExprNodes.ListNode(node.pos, args=[], constant_result=[])