Enable model/dataset versioning in test.sh

## Type of Issue

Select the type of issue:
- [ ] Bug report (to report a bug)
- [x] Feature request (to request an additional feature)
- [ ] Tracker (I am just using this as a tracker)
- [ ] Refactor request
- [ ] Documentation Ask

## Description

We should enable model/dataset versioning in test.sh so that when we run unittest locally it will pull the corresponding model. We can also pass some arguments (e.g. --debug) so that we can test with different models just like what we did in agent.