Add tests for GO statement parsing