Add unit tests for type checker