Improve test coverage for models

Summary

Add unit tests for models.

Reasoning

Improve overall test coverage.