Commit1838f1c

committed

Add test for syntax error (and fix colorization)

1 parente4a0717 commit1838f1cCopy full SHA for 1838f1c

File tree

+36

-5

lines changed

+36

-5

lines changed

Lines changed: 21 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -4284,6 +4284,27 @@ def bar():`
`4284`	`4284`	`self.assertIn(boldr+",4)"+reset,lines)`
`4285`	`4285`	`self.assertIn(red+"bar"+reset+boldr+"()"+reset,lines)`
`4286`	`4286`
	`4287`	`+deftest_colorized_syntax_error(self):`
	`4288`	`+try:`
	`4289`	`+compile("a $ b","<string>","exec")`
	`4290`	`+exceptSyntaxErrorase:`
	`4291`	`+exc=traceback.TracebackException.from_exception(`
	`4292`	`+e,capture_locals=True`
	`4293`	`+ )`
	`4294`	`+actual="".join(exc.format(colorize=True))`
	`4295`	`+red=traceback._ANSIColors.RED`
	`4296`	`+magenta=traceback._ANSIColors.MAGENTA`
	`4297`	`+boldm=traceback._ANSIColors.BOLD_MAGENTA`
	`4298`	`+boldr=traceback._ANSIColors.BOLD_RED`
	`4299`	`+reset=traceback._ANSIColors.RESET`
	`4300`	`+expected="".join([`
	`4301`	`+f' File{magenta}"<string>"{reset}, line{magenta}1{reset}\n',`
	`4302`	`+f' a{boldr}${reset} b\n',`
	`4303`	`+f'{boldr}^{reset}\n',`
	`4304`	`+f'{boldm}SyntaxError{reset}:{magenta}invalid syntax{reset}\n']`
	`4305`	`+ )`
	`4306`	`+self.assertIn(expected,actual)`
	`4307`	`+`
`4287`	`4308`	`deftest_colorized_traceback_is_the_default(self):`
`4288`	`4309`	`deffoo():`
`4289`	`4310`	`1/0`

Lines changed: 15 additions & 5 deletions

Original file line number	Diff line number	Diff line change
`@@ -1310,22 +1310,32 @@ def _format_syntax_error(self, stype, **kwargs):`
`1310`	`1310`	`# colorize from colno to end_colno`
`1311`	`1311`	`ltext= (`
`1312`	`1312`	`ltext[:colno]+`
`1313`		`-_ANSIColors.RED+ltext[colno:end_colno]+_ANSIColors.RESET+`
	`1313`	`+_ANSIColors.BOLD_RED+ltext[colno:end_colno]+_ANSIColors.RESET+`
`1314`	`1314`	`ltext[end_colno:]`
`1315`	`1315`	`)`
`1316`		`-start_color=_ANSIColors.RED`
	`1316`	`+start_color=_ANSIColors.BOLD_RED`
`1317`	`1317`	`end_color=_ANSIColors.RESET`
`1318`	`1318`	`yield' {}\n'.format(ltext)`
`1319`		`-yield' {}{}{}{}'.format(`
	`1319`	`+yield' {}{}{}{}\n'.format(`
`1320`	`1320`	`"".join(caretspace),`
`1321`	`1321`	`start_color,`
`1322`		`- ('^'* (end_colno-colno)+"\n"),`
	`1322`	`+ ('^'* (end_colno-colno)),`
`1323`	`1323`	`end_color,`
`1324`	`1324`	`)`
`1325`	`1325`	`else:`
`1326`	`1326`	`yield' {}\n'.format(ltext)`
`1327`	`1327`	`msg=self.msgor"<no detail available>"`
`1328`		`-yield"{}: {}{}\n".format(stype,msg,filename_suffix)`
	`1328`	`+ifcolorize:`
	`1329`	`+yield"{}{}{}: {}{}{}{}\n".format(`
	`1330`	`+_ANSIColors.BOLD_MAGENTA,`
	`1331`	`+stype,`
	`1332`	`+_ANSIColors.RESET,`
	`1333`	`+_ANSIColors.MAGENTA,`
	`1334`	`+msg,`
	`1335`	`+_ANSIColors.RESET,`
	`1336`	`+filename_suffix)`
	`1337`	`+else:`
	`1338`	`+yield"{}: {}{}\n".format(stype,msg,filename_suffix)`
`1329`	`1339`
`1330`	`1340`	`defformat(self,,chain=True,_ctx=None,*kwargs):`
`1331`	`1341`	`"""Format the exception.`

Comments

(0)