How to Catch and Print Assertion Errors in Python

[Sunwoo Bae]

I would like to make output only read:

Traceback (most recent call last):
AssertionError

... without giving the File "<input>", line 1, in <module>

 

[Sunwoo Bae]

 

[pbuk]

Why? The line number is there to help you debug your code. If this is to show a message to an end user then you should be using try...except, not just letting the program fail.

 

[pbuk]

Oh its for unit testing? You should use assertRaises in your test instead of testing the crash dump output.

 

[Sunwoo Bae]

Oh its for unit testing? You should use assertRaises in your test instead of testing the crash dump output.

can you tell me what assertRaises exactly do and how to use it? I have never heard about it :(

 

[pbuk]

Hmmm, looking again I am not sure you are using a unit test framework. Is this an online tutorial? If so, stick with my first answer: catch the assertion and print its message. You can look up how to catch the failing assertion with try...except and how to get the message string from an AssertionError in a Python reference if you haven't covered them already (have you skipped some of the tutorial?).