complete the implementation of loopy.transform.data.rename_argument

- handle cases with argument in domain
- argument as a part of a variables shape expr.
- added a test to check such use cases