Added a bunch of tests for the models.

Signed-off-by: Tushar Narayan <tusharnarayan@gmail.com>