Add basic set of tests for model generator