@@ -94,18 +94,22 @@ def _underline_token_in_template(
94
94
95
95
template_before_token = token .template [: token .start_position ]
96
96
if skipped_lines := template_before_token .count ("\n " ) - lines_around :
97
- template_before_token = (
98
- f"{ cls ._skipped_lines_message (skipped_lines )} \n "
99
- + "\n " .join (template_before_token .split ("\n " )[- (lines_around + 1 ) :])
97
+ template_before_token = "\n " .join (
98
+ template_before_token .split ("\n " )[- (lines_around + 1 ) :]
100
99
)
101
100
101
+ if 0 < skipped_lines :
102
+ template_before_token = f"{ cls ._skipped_lines_message (skipped_lines )} \n { template_before_token } "
103
+
102
104
template_after_token = token .template [token .end_position :]
103
105
if skipped_lines := template_after_token .count ("\n " ) - lines_around :
104
- template_after_token = (
105
- "\n " .join (template_after_token .split ("\n " )[: (lines_around + 1 )])
106
- + f"\n { cls ._skipped_lines_message (skipped_lines )} "
106
+ template_after_token = "\n " .join (
107
+ template_after_token .split ("\n " )[: (lines_around + 1 )]
107
108
)
108
109
110
+ if 0 < skipped_lines :
111
+ template_after_token += f"\n { cls ._skipped_lines_message (skipped_lines )} "
112
+
109
113
lines_before_line_with_token = template_before_token .rsplit ("\n " , 1 )[0 ]
110
114
111
115
line_with_token = (
@@ -122,7 +126,7 @@ def _underline_token_in_template(
122
126
123
127
lines_after_line_with_token = template_after_token .split ("\n " , 1 )[- 1 ]
124
128
125
- return "\n " .join (
129
+ return "\n " + " \n " .join (
126
130
[
127
131
lines_before_line_with_token ,
128
132
line_with_token ,
0 commit comments