#NEWLINE# def test_on_failure_callback_raises_error( self ) : #NEWLINE# #INDENT# exec_graph = ExecutionGraph ( [ self . job ( 'A' , raising_fn , [ ] , on_failure = raising_fn ) ] ) #NEWLINE# with self . assertRaises ( ExecutionFailure ) as cm : #NEWLINE# #INDENT# self . execute ( self ) #NEWLINE# #UNINDENT# self . assertEqual ( "Error in on_failure for A: I'm an error" , str ( cm . exception ) )