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)