fix tests/test_unsupported.py dict_del

py2many · Aug 11, 2024 · affbf1b · affbf1b
1 parent d2118d4
commit affbf1b
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/pyd/transpiler.py b/pyd/transpiler.py
@@ -473,9 +473,17 @@ def visit_Tuple(self, node) -> str:
         return "({0})".format(elts)
 
     def visit_Delete(self, node) -> str:
-        self._usings.add("core.memory")
         target = node.targets[0]
         target_str = self.visit(target)
+        if len(node.targets) == 1 and isinstance(target, ast.Subscript):
+            self._usings.add("std.algorithm")
+            # special handling to delete element from container, e.g.
+            # tests/test_unsupported.py dict_del: "a = {1: 1}; del a[1]; assert not a",
+            value = self.visit(target.value)
+            index = self.visit(target.slice)
+            return "{0}.remove({1});".format(value, index)
+
+        self._usings.add("core.memory")
         return (
             f"destroy!true({target_str}); core.memory.GC.free(cast(void*){target_str});"
         )