if __name__ == '__main__':
from argparse import ArgumentParser
parser = ArgumentParser()
parser.add_argument('--rounds', type=int, default=10)
parser.add_argument('--verbosity', type=int, default=1)
parser.add_argument('--generations', type=int, default=1)
parser.add_argument('--mult', type=int, default=1, help='multiplier of distribution')
parser.add_argument('--dist', help='distribution', default=None)
args = parser.parse_args()
if args.dist:
dist = literal_eval(args.dist)
if 'both' in dist:
for sex in ['male', 'female']:
if sex in dist:
dist[sex].update(dist['both'])
else:
dist[sex] = dist['both']
print(repr(dist))
tourn = Tournament(rounds=args.rounds)
tourn.verbosity = args.verbosity
for sex in ['male', 'female']:
for name in dist[sex]:
for ii in range(dist[sex][name] * args.mult):
tourn.add_player(create_simple_player(sex=sex, name=name))
for gen in range(args.generations):
tourn.resolve()
tourn.reproduce()
print()
tourn.show_distribution()
print()
