Uso conveniente de assert em Python

Recentemente, aprendi o uso fácil de assert, então gostaria de fazer um memorando.

quando o resultado é False, assert retorna AssertionError.

>>> assert False
AssertionError:

Se você conectar o assert com vírgula (,) e string, então a string será exibida após o erro como abaixo.

>>> assert False, "it's wrong"
AssertionError: it's wrong

Portanto, se você deseja declarar um argumento em uma função, pode facilmente escrever uma verificação simples:

def do_something(mode="train"):
    assert mode in ['train', 'test'], "mode only takes either train or test"

Ao escrever assim, um simples cheque se torna um pedaço de bolo.

>>> do_something(mode='something')
AssertionError: mode only takes either train or test.

Por hoje é isso.