PyFile:AwaitOnTheSameLineAsFor.py
  PyFunction('y')
    PsiElement(Py:ASYNC_KEYWORD)('async')
    PsiWhiteSpace(' ')
    PsiElement(Py:DEF_KEYWORD)('def')
    PsiWhiteSpace(' ')
    PsiElement(Py:IDENTIFIER)('y')
    PyParameterList
      PsiElement(Py:LPAR)('(')
      PyNamedParameter('a1')
        PsiElement(Py:IDENTIFIER)('a1')
      PsiElement(Py:COMMA)(',')
      PsiWhiteSpace(' ')
      PyNamedParameter('a2')
        PsiElement(Py:IDENTIFIER)('a2')
      PsiElement(Py:RPAR)(')')
    PsiElement(Py:COLON)(':')
    PsiWhiteSpace('\n    ')
    PyStatementList
      PyForStatement
        PyForPart
          PsiElement(Py:FOR_KEYWORD)('for')
          PsiWhiteSpace(' ')
          PyTargetExpression: i
            PsiElement(Py:IDENTIFIER)('i')
          PsiWhiteSpace(' ')
          PsiElement(Py:IN_KEYWORD)('in')
          PsiWhiteSpace(' ')
          PyCallExpression: range
            PyReferenceExpression: range
              PsiElement(Py:IDENTIFIER)('range')
            PyArgumentList
              PsiElement(Py:LPAR)('(')
              PyNumericLiteralExpression
                PsiElement(Py:INTEGER_LITERAL)('10')
              PsiElement(Py:RPAR)(')')
          PsiElement(Py:COLON)(':')
          PsiWhiteSpace(' ')
          PyStatementList
            PyExpressionStatement
              PyCallExpression: a1.foo
                PyReferenceExpression: foo
                  PyReferenceExpression: a1
                    PsiElement(Py:IDENTIFIER)('a1')
                  PsiElement(Py:DOT)('.')
                  PsiElement(Py:IDENTIFIER)('foo')
                PyArgumentList
                  PsiElement(Py:LPAR)('(')
                  PyPrefixExpression
                    PsiElement(Py:AWAIT_KEYWORD)('await')
                    PsiWhiteSpace(' ')
                    PyCallExpression: a1.x
                      PyReferenceExpression: x
                        PyReferenceExpression: a1
                          PsiElement(Py:IDENTIFIER)('a1')
                        PsiElement(Py:DOT)('.')
                        PsiElement(Py:IDENTIFIER)('x')
                      PyArgumentList
                        PsiElement(Py:LPAR)('(')
                        PyReferenceExpression: a2
                          PsiElement(Py:IDENTIFIER)('a2')
                        PsiElement(Py:RPAR)(')')
                  PsiElement(Py:RPAR)(')')