Ambiente de prática

Teste um código gerado por IA

Abaixo está uma função que uma IA poderia gerar ao receber o pedido “calcule a média de uma lista”. Copie o código e os testes para um notebook no Google Colab e execute.

1. Prepare o ambiente

Célula do Colab

!pip install pytest

2. O código gerado

media.py

def media(numeros: list[float]) -> float:
    return sum(numeros) / len(numeros)

3. Os testes

test_media.py

import pytest
from media import media

@pytest.mark.parametrize("entrada, esperado", [
    ([2, 4, 6], 4),
    ([10], 10),
    ([-1, 1], 0),
])
def test_media_happy_path(entrada, esperado):
    assert media(entrada) == esperado

def test_media_lista_vazia():
    with pytest.raises(ZeroDivisionError):
        media([])

Célula do Colab

!pytest -q

Para refletir

  • O teste da lista vazia passa, mas o requisito diz que a função deveria lançar um erro? Ou retornar 0?
  • Que outros edge cases a IA não considerou? Pense em valores não numéricos.
  • Reescreva o prompt para o Gemini incluindo o comportamento esperado para entradas inválidas.

Pratique no notebook das aulas

O notebook do Google Colab usado nas gravações está em Materiais de apoio. Salve uma cópia no seu Drive e adicione os novos testes nele.