"""Tests para cp_provincia_es."""

import sys
import os

sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))

from cp_provincia_es import cp_provincia_es


def test_cp_completo_retorna_provincia():
    """cp completo retorna provincia"""
    assert cp_provincia_es("28001") == "Madrid"


def test_prefijo_2_digitos_retorna_provincia():
    """prefijo 2 digitos retorna provincia"""
    assert cp_provincia_es("28") == "Madrid"


def test_primer_prefijo_01_retorna_alava():
    """primer prefijo 01 retorna Alava"""
    assert cp_provincia_es("01") == "Álava"


def test_cp_desconocido_retorna_none():
    """cp desconocido retorna None"""
    assert cp_provincia_es("99") is None


def test_cp_entero_completo():
    assert cp_provincia_es(28001) == "Madrid"


def test_cp_ceuta():
    assert cp_provincia_es("51001") == "Ceuta"


def test_cp_melilla():
    assert cp_provincia_es("52") == "Melilla"


def test_cp_barcelona():
    assert cp_provincia_es("08") == "Barcelona"
