Writing tests for lexer