wip

Author: unho

stdnum/vatin.py

@@ -66,12 +66,11 @@ def _get_cc_module(cc):
     # Greece uses a "wrong" country code, special case for Northern Ireland
     cc = cc.lower().replace('el', 'gr').replace('xi', 'gb')
     if not _country_code_re.match(cc):
-        raise ImportError('Wrong "{}" country code.'.format(cc))
+        raise ImportError('Wrong country code.')
     if cc not in _country_modules:
         _country_modules[cc] = get_cc_module(cc, 'vat')
     if not _country_modules[cc]:
-        raise ImportError('Missing validation for "{}" country code.'
-                          ''.format(cc))
+        raise ImportError('Missing validation for country code.')
     return _country_modules[cc]
 
 

tests/test_vatin.doctest

@@ -61,23 +61,23 @@ Try validating not specifying a country:
 >>> vatin.validate('')
 Traceback (most recent call last):
     ...
-ImportError: ...
+ImportError: Wrong country code.
 >>> vatin.validate('00')
 Traceback (most recent call last):
     ...
-ImportError: ...
+ImportError: Wrong country code.
 
 
 Try to validate for countries with no VAT validation:
 
 >>> vatin.validate('XX')
 Traceback (most recent call last):
     ...
-ImportError: ...
+ImportError: Missing validation for country code.
 >>> vatin.validate('US')
 Traceback (most recent call last):
     ...
-ImportError: ...
+ImportError: Missing validation for country code.
 
 
 Check is_valid for several scenarios: