O comando assert

Como o assert funciona

O comando assert recebe uma condição, exatamente do mesmo tipo que você usa em um if. A ideia é simples: você afirma que aquela condição é verdadeira naquele ponto do programa. Se ela realmente for, o Python não faz nada e a execução continua. Se ela for falsa, o Python entende que algo saiu do esperado e interrompe o programa levantando um erro chamado AssertionError. Em outras palavras, assert é o jeito de dizer eu garanto que aqui isto é verdade, e deixar o Python cobrar essa garantia.

def dobro(n):
    return n * 2

# Afirma o que deveria ser verdade:
assert dobro(5) == 10
assert dobro(0) == 0
assert dobro(-3) == -6

print("Todos os asserts passaram!")
# Saida: Todos os asserts passaram!

Três afirmações verdadeiras: nenhuma reclama e o print no final é executado.

Repare no comportamento silencioso quando tudo está certo. Os três asserts acima passam, o programa não imprime nada sobre eles e chega ao print final. Esse silêncio é proposital: teste que passa não precisa falar, teste que falha precisa gritar. Agora veja o que acontece quando a afirmação é falsa, por exemplo se alguém quebrar a função dobro trocando a multiplicação por soma.

def dobro(n):
    return n + 2   # bug: era n * 2

assert dobro(5) == 10
# Saida:
# Traceback (most recent call last):
#   ...
# AssertionError

Com a função quebrada, dobro(5) devolve 7, a condição fica falsa e o assert dispara.

A mensagem que explica a falha

Um AssertionError sem explicação diz que algo deu errado, mas não diz o quê. Por isso o assert aceita uma segunda parte, depois de uma vírgula: uma mensagem que aparece junto do erro. Use-a para descrever o que se esperava, de preferência mostrando o valor que veio errado. Uma boa mensagem transforma um teste que falha de um enigma em um recado direto, e economiza minutos de investigação.

def media(valores):
    return sum(valores) / len(valores)

resultado = media([2, 4, 6])
esperado = 4

assert resultado == esperado, f"esperava {esperado}, veio {resultado}"
print("Media conferida:", resultado)
# Saida: Media conferida: 4.0

A mensagem com f-string mostra o esperado e o obtido, caso a condição falhe.

O padrão que aparece aqui vai se repetir o curso inteiro: você calcula o resultado obtido, escreve o resultado esperado e afirma que os dois são iguais. Essa é a espinha dorsal de qualquer teste, com assert, com unittest ou com pytest. Só muda a roupagem. Guardar bem esse trio, obtido, esperado e a comparação, é entender testes de verdade.

Testando o entendimento

Vale reforçar a diferença entre os dois usos de uma condição que você já conhece. O if pergunta e desvia: se for verdade, faz uma coisa, se não, faz outra. O assert não pergunta, ele exige: se for verdade, segue em frente, se não, para tudo com um erro. Em um teste, você usa assert porque não quer tratar o caso falso com elegância, você quer ser avisado em alto e bom som de que a garantia foi quebrada. É essa a mentalidade que unittest e pytest, nas próximas aulas, apenas organizam e deixam mais confortável.