Fix location information

Author: certik

src/lpython/parser/parser.yy

@@ -315,7 +315,8 @@ script_unit
     ;
 
 statements
-    : TK_INDENT statements1 TK_DEDENT { $$ = $2; LLOC(@$,$2); }
+    : TK_INDENT statements1 TK_DEDENT { $$ = $2; LLOC(@$,@2);
+        @$.first = @2.first; }
     ;
 
 sep_statements
@@ -334,9 +335,9 @@ statements1
     ;
 
 single_line_statements
-    : single_line_multi_statements TK_NEWLINE { $$ = $1; }
+    : single_line_multi_statements TK_NEWLINE { $$ = $1; LLOC(@$,@1); }
     | single_line_multi_statements TK_EOLCOMMENT { $$ = $1; }
-    | single_line_statement TK_NEWLINE { $$ = A2LIST(p.m_a, $1); }
+    | single_line_statement TK_NEWLINE { $$ = A2LIST(p.m_a, $1); LLOC(@$,@1); }
     | single_line_statement TK_SEMICOLON TK_NEWLINE { $$ = A2LIST(p.m_a, $1); }
     | single_line_statement TK_SEMICOLON TK_EOLCOMMENT { $$ = A2LIST(p.m_a, $1); }
     | single_line_statement TK_EOLCOMMENT { $$ = A2LIST(p.m_a, $1); }
@@ -358,10 +359,10 @@ type_ignore_sep
     ;
 
 statement
-    : single_line_statement sep { $$ = $1; }
+    : single_line_statement sep { $$ = $1; LLOC(@$,@1); }
     | single_line_statement type_ignore_sep { $$ = $1; }
     | multi_line_statement
-    | multi_line_statement sep { $$ = $1; }
+    | multi_line_statement sep { $$ = $1; LLOC(@$,@1); }
     | multi_line_statement type_ignore_sep { $$ = $1; }
     ;