def testRawComment(self): lex = Lexer() lex.setSrc('//sdfsdff\n//asdasd') self.assertEqual(lex.getNext(), (TOK.SINGLE_COMMENT,'sdfsdff')) lex.getNext() self.assertEqual(lex.getNext(), (TOK.SINGLE_COMMENT,'asdasd')) lex.setSrc("""//sdfdff ttttt""") self.assertEqual(lex.getNext(), (TOK.SINGLE_COMMENT,'sdfdff')) lex.setSrc('/*asdasd*/') self.assertEqual(lex.getNext(), (TOK.MULTI_COMMENT, 'asdasd')) lex.setSrc('/*asd\nasd*/123123') self.assertEqual(lex.getNext(), (TOK.MULTINL_COMMENT, 'asd\nasd'))