How to use Python’s assert keyword properly