Added unit test for model module.