def test_encode_birthplace_invalid_arguments(self):
with self.assertRaises(ValueError):
codicefiscale.encode_birthplace(None)
with self.assertRaises(ValueError):
codicefiscale.encode_birthplace('Area 51')
def test_encode_birthplace_italy(self):
data = [
{
'input': 'Torino, Italy',
'result': 'L219',
},
{
'input': 'Torino (TO), Italy',
'result': 'L219',
},
{
'input': 'Torino (TO)',
'result': 'L219',
},
{
'input': 'Torino',
'result': 'L219',
},
{
'input': 'L219',
'result': 'L219',
},
]
for obj in data:
# with self.subTest(obj=obj):
self.assertEqual(codicefiscale.encode_birthplace(obj['input']),
obj['result'])
def test_encode_birthplace_foreign_country(self):
data = [
{
'input': 'Lettonia',
'result': 'Z145',
},
{
'input': 'Giappone',
'result': 'Z219',
},
{
'input': 'Marocco',
'result': 'Z330',
},
]
for obj in data:
# with self.subTest(obj=obj):
self.assertEqual(codicefiscale.encode_birthplace(obj['input']),
obj['result'])
