recurse

Author: BobTheBuidler

mypyc/codegen/literals.py

@@ -141,7 +141,20 @@ def encoded_tuple_values(self) -> list[str]:
         return self._encode_collection_values(self.tuple_literals)
 
     def encoded_frozenset_values(self) -> list[str]:
-        return self._encode_collection_values(self.frozenset_literals)
+        def sort_frozenset_recursive(frozen: frozenset[object]) -> list[object]:
+            # even though frozensets are not sorted in python, we need to sort the items here
+            # to improve the determinism of the generated C file, making it easier to compare
+            # differences in the C files generated by different versions of your code.
+            sort_helper = {repr(v) + type(v).__name__: v for v in value}
+            items = []
+            for key in sorted(sort_helper):
+                v = sort_helper[key]
+                if isinstance(v, frozenset):
+                    v = sort_frozenset_items(v)
+                items.append(v)
+            return items
+
+        return self._encode_collection_values({sort_frozenset_items(fr): i for fr, i in self.frozenset_literals.items()})
 
     def _encode_collection_values(
         self, values: dict[tuple[object, ...], int] | dict[frozenset[object], int]
@@ -165,15 +178,7 @@ def _encode_collection_values(
         for i in range(count):
             value = value_by_index[i]
             result.append(str(len(value)))
-            if isinstance(value, frozenset):
-                # even though frozensets are not sorted in python, we need to sort the items here
-                # to improve the determinism of the generated C file, making it easier to compare
-                # differences in the C files generated by different versions of your code.
-                sort_helper = {str(v) + type(v).__name__: v for v in value}
-                items = tuple(sort_helper[key] for key in sorted(sort_helper))
-            else:
-                items = value
-            for item in items:
+            for item in value:
                 assert _is_literal_value(item)
                 index = self.literal_index(item)
                 result.append(str(index))