Assert statements in Python