Add more trivial test cases for lexer