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
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:
assert mode in ['train', 'test'], "mode only takes either train or test"
By writing like this, a simple check becomes a piece of cake.
AssertionError: mode only takes either train or test.
That's it for today.