Move unit tests to model specs