Fixed write in augmented assignments

src/dynapyt/__init__.py

@@ -1,2 +1,2 @@
 """DynaPyt"""
-__version__ = "1.2.1"
+__version__ = "1.3.0"

src/dynapyt/runtime.py

@@ -155,8 +155,35 @@ def _aug_assign_(dyn_ast, iid, left, opr, right):
         "binary_operation", dyn_ast, iid, operator[opr][:-6], left, right, None
     )
     call_if_exists(snake(operator[opr][:-6]), dyn_ast, iid, left, right, None)
-    call_if_exists("memory_access", dyn_ast, iid, right)
-    call_if_exists("write", dyn_ast, iid, [left], right)
+    eval_left = left()
+    if opr == 0:
+        new_val = eval_left + right
+    elif opr == 1:
+        new_val = eval_left & right
+    elif opr == 2:
+        new_val = eval_left | right
+    elif opr == 3:
+        new_val = eval_left ^ right
+    elif opr == 4:
+        new_val = eval_left / right
+    elif opr == 5:
+        new_val = eval_left // right
+    elif opr == 6:
+        new_val = eval_left << right
+    elif opr == 7:
+        new_val = eval_left @ right
+    elif opr == 8:
+        new_val = eval_left % right
+    elif opr == 9:
+        new_val = eval_left * right
+    elif opr == 10:
+        new_val = eval_left**right
+    elif opr == 11:
+        new_val = eval_left >> right
+    elif opr == 12:
+        new_val = eval_left - right
+    call_if_exists("memory_access", dyn_ast, iid, new_val)
+    call_if_exists("write", dyn_ast, iid, [left], new_val)
     result_high = call_if_exists(
         "augmented_assignment", dyn_ast, iid, left, operator[opr], right
     )

tests/trace_single_hook/write/expected.txt

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

tests/trace_single_hook/write/program.py

@@ -1,3 +1,5 @@
 a = 10
 b = "test"
 c = a * b
+b += "ing"
+a *= 3