---input---
def foo():
    ur"""unicode-raw"""

def bar():
    u"""unicode"""

def baz():
    r'raw'

def zap():
    """docstring"""

---tokens---
'def'         Keyword
' '           Text
'foo'         Name.Function
'('           Punctuation
')'           Punctuation
':'           Punctuation
'\n'          Text

'    '        Text
'ur'          Literal.String.Affix
'"""unicode-raw"""' Literal.String.Doc
'\n'          Text

'\n'          Text

'def'         Keyword
' '           Text
'bar'         Name.Function
'('           Punctuation
')'           Punctuation
':'           Punctuation
'\n'          Text

'    '        Text
'u'           Literal.String.Affix
'"""unicode"""' Literal.String.Doc
'\n'          Text

'\n'          Text

'def'         Keyword
' '           Text
'baz'         Name.Function
'('           Punctuation
')'           Punctuation
':'           Punctuation
'\n'          Text

'    '        Text
'r'           Literal.String.Affix
"'"           Literal.String.Single
'raw'         Literal.String.Single
"'"           Literal.String.Single
'\n'          Text

'\n'          Text

'def'         Keyword
' '           Text
'zap'         Name.Function
'('           Punctuation
')'           Punctuation
':'           Punctuation
'\n'          Text

'    '        Text
'"""docstring"""' Literal.String.Doc
'\n'          Text
