Add some more tests for models related init

Test Plan:
- Run tests