def test_dame_gender_guess_surname(self):
g = Gender()
self.assertEqual(g.guess_surname("Smith", "us"), [True, 2376206])
self.assertEqual(g.guess_surname("Serrano", "es"), [True, 131829])
import os
import re
import argparse
parser = argparse.ArgumentParser()
parser.add_argument("surname", help="display the gender")
parser.add_argument('--total',
required=True,
default="ine",
choices=['ine', 'es', 'us'])
parser.add_argument('--version', action='version', version='0.1')
parser.add_argument('--verbose', default=False, action="store_true")
args = parser.parse_args()
results = []
g = Gender()
v = g.guess_surname(args.surname, args.total)
if (v[0] == True):
if ((args.total == 'es') or (args.total == 'ine')):
print("There are %s people using %s in Spain" % (v[1], args.surname))
elif (args.total == 'us'):
print("There are %s people using %s in United States of America" %
(v[1], args.surname))
else:
if ((args.total == 'es') or (args.total == 'ine')):
print("There are not people using %s in Spain" % args.surname)
elif (args.total == 'us'):
print("There are not people using %s in United States of America" %
args.surname)