Recently, I learnt the easy use of assert, so I'd like to make a memo.
when the result is False, assert returns AssertionError.
>>> assert False AssertionError:
If you connect the assert with comma ( , ) and string, then the string gets displayed after the error as below.
>>> assert False, "it's wrong" AssertionError: it's wrong
So, if you want to assert an argument in a function, you can easily write a simple check:
def do_something(mode="train"): assert mode in ['train', 'test'], "mode only takes either train or test"
By writing like this, a simple check becomes a piece of cake.
>>> do_something(mode='something') AssertionError: mode only takes either train or test.
That's it for today.