Adds tests to models

@gsantikian @chrisscherer please merge these before changing the models