StackRduino
diff --git a/‎py/objgenerator.c‎
Lines changed: 1 addition & 0 deletions b/‎py/objgenerator.c‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎py/vm.c‎
Lines changed: 2 additions & 1 deletion b/‎py/vm.c‎
Lines changed: 2 additions & 1 deletion
@@ -99,6 +99,7 @@ mp_vm_return_kind_t mp_obj_gen_resume(mp_obj_t self_in, mp_obj_t send_value, mp_
     assert(MP_OBJ_IS_TYPE(self_in, &mp_type_gen_instance));
     mp_obj_gen_instance_t *self = MP_OBJ_TO_PTR(self_in);
     if (self->code_state.ip == 0) {
+        // Trying to resume already stopped generator
         *ret_val = MP_OBJ_STOP_ITERATION;
         return MP_VM_RETURN_NORMAL;
     }
 
@@ -1144,7 +1144,8 @@ unwind_jump:;
                     if (ret_kind == MP_VM_RETURN_NORMAL) {
                         // Pop exhausted gen
                         sp--;
-                        if (ret_value == MP_OBJ_NULL) {
+                        // TODO: When ret_value can be MP_OBJ_NULL here??
+                        if (ret_value == MP_OBJ_NULL || ret_value == MP_OBJ_STOP_ITERATION) {
                             // Optimize StopIteration
                             // TODO: get StopIteration's value
                             PUSH(mp_const_none);
Original file line number	Diff line number	Diff line change
`@@ -99,6 +99,7 @@ mp_vm_return_kind_t mp_obj_gen_resume(mp_obj_t self_in, mp_obj_t send_value, mp_`
`99`	`99`	`assert(MP_OBJ_IS_TYPE(self_in, &mp_type_gen_instance));`
`100`	`100`	`mp_obj_gen_instance_t *self = MP_OBJ_TO_PTR(self_in);`
`101`	`101`	`if (self->code_state.ip == 0) {`
	`102`	`+ // Trying to resume already stopped generator`
`102`	`103`	`*ret_val = MP_OBJ_STOP_ITERATION;`
`103`	`104`	`return MP_VM_RETURN_NORMAL;`
`104`	`105`	`}`