def test_comparison(self):
     source = "2 < 3.5"
     
     lexer.input(source)
     tokens = [token.type for token in lexer]
     
     expected = ['INTEGER', 'COMP_LT', 'FLOAT']
     self.assertListEqual(expected, tokens)
 def test_operators(self):
     source = "!+*-/"
     
     lexer.input(source)
     tokens = [token.type for token in lexer]
     
     expected = ['!', '+', '*', '-', '/']
     self.assertListEqual(expected, tokens)
 def test_brackets(self):
     source = "[{}]()"
     
     lexer.input(source)
     tokens = [token.type for token in lexer]
     
     expected = ['[', '{', '}', ']', '(', ')']
     self.assertListEqual(expected, tokens)
 def test_function(self):
     src = """[x,y->t=1;z+1];"""
     lexer.input(src)
     self.assertEquals(14, len(list(lexer)))