Add unit tests for models-base