Fixed issue with tuples in assignments

src/dynapyt/instrument/CodeInstrumenter.py

@@ -1107,6 +1107,12 @@ def leave_Assign(self, original_node, updated_node):
             val_arg = cst.Arg(value=updated_node.value.value.item)
         elif m.matches(updated_node.value, m.Yield()):
             val_arg = cst.Arg(value=updated_node.value.value)
+        elif m.matches(updated_node.value, m.Tuple()):
+            val_arg = cst.Arg(
+                value=updated_node.value.with_changes(
+                    lpar=[cst.LeftParen()], rpar=[cst.RightParen()]
+                )
+            )
         else:
             val_arg = cst.Arg(value=updated_node.value)
         left_arg = cst.Arg(
@@ -1149,6 +1155,12 @@ def leave_AnnAssign(self, original_node, updated_node):
         iid_arg = cst.Arg(value=cst.Integer(value=str(iid)))
         if m.matches(updated_node.value, m.Yield()):
             val_arg = cst.Arg(value=updated_node.value.value)
+        elif m.matches(updated_node.value, m.Tuple()):
+            val_arg = cst.Arg(
+                value=updated_node.value.with_changes(
+                    lpar=[cst.LeftParen()], rpar=[cst.RightParen()]
+                )
+            )
         else:
             val_arg = cst.Arg(value=updated_node.value)
         left_arg = cst.Arg(

tests/trace_single_hook/write/expected.txt

@@ -4,4 +4,6 @@ Writing test to [('b',)]
 Writing testtesttesttesttesttesttesttesttesttest to [('c',)]
 Writing testing to [('b',)]
 Writing 30 to [('a',)]
+Writing ('test',) to [('d',)]
+Writing ('test',) to [('e',)]
 end execution

tests/trace_single_hook/write/program.py

@@ -3,3 +3,5 @@
 c = a * b
 b += "ing"
 a *= 3
+d = "test",
+e: str = "test",