Add tests for semantic model