update test function for update_w_w0_v4

1. do minor changes on annotation in this funciton