Exemplo de Solução
O seguinte programa mostra uma maneira de resolver este problema. Essa não é a única forma possível de solucioná-lo.
# Direitos Autorais 2020, Universidade Brigham Young-Idaho. Todos os direitos reservados.
"""
Verifica se as funções extrair_cidade, extrair_estado
e extrair_cep funcionam corretamente.
"""
from endereco import extrair_cidade, extrair_estado, extrair_cep
import pytest
def test_extrair_cidade():
"""Verifica se a função extrair_cidade retorna os resultados corretos.
Parâmetros: nenhum
Retorno: nada
"""
assert extrair_cidade("Rua das Flores, 123, Centro, Curitiba - PR, 80000-000") == "Curitiba"
assert extrair_cidade("Avenida Paulista, 1000, Bela Vista, São Paulo - SP, 01310-100") == "São Paulo"
def test_extrair_estado():
"""Verifica se a função extrair_estado retorna os resultados corretos.
Parâmetros: nenhum
Retorno: nada
"""
assert extrair_estado("Rua das Flores, 123, Centro, Curitiba - PR, 80000-000") == "PR"
assert extrair_estado("Avenida Paulista, 1000, Bela Vista, São Paulo - SP, 01310-100") == "SP"
def test_extrair_cep():
"""Verifica se a função extrair_cep retorna os resultados corretos.
Parâmetros: nenhum
Retorno: nada
"""
assert extrair_cep("Rua das Flores, 123, Centro, Curitiba - PR, 80000-000") == "80000-000"
assert extrair_cep("Avenida Paulista, 1000, Bela Vista, São Paulo - SP, 01310-100") == "01310-100"
# Chama a função principal do pytest para que
# os testes neste arquivo sejam executados.
pytest.main(["-v", "--tb=line", "-rN", __file__])