def test_country_invalid(): with pytest.raises(InvalidCountry): country_prefix('dkk')
def test_country_prefix(): assert country_prefix('dk') == 45 assert country_prefix('DK') == 45 assert country_prefix('Dk') == 45 assert country_prefix('us') == 1