[FEATURE] Add unit tests for the models.
Merge pull request #4 from braingourmets/model-tests

[FEATURE] Add unit tests for the models.