def test_number_is_int(self):
lexer = Lexer('3')
result = lexer.number()
assert result == Token('INTEGER', 3)
def test_invalid_character(self):
lexer = Lexer('bob')
with pytest.raises(SyntaxError) as ex:
lexer.get_next_token()
assert 'Invalid character' == str(ex.value)
def test_number_is_float(self):
lexer = Lexer('3.2')
result = lexer.number()
assert result == Token('FLOAT', 3.2)
