add @assign test cases, fix layout issue in @assign, @break and @continue