Adjust main to print number of wins and games played

This commit is contained in:
2024-01-31 17:42:19 +01:00
parent 793d83b943
commit 13403b74f3

14
main.py
View File

@@ -101,13 +101,15 @@ def test_evaluation():
evaluator = simulation.Evaluation(a, strat1, b, strat2, limit) evaluator = simulation.Evaluation(a, strat1, b, strat2, limit)
results = evaluator.run(n, proc) results = evaluator.run(n, proc)
a_results = len(list(filter(lambda x: x.winner == simulation.Winner.Engine_A, results))) / len(results) * 100 games_played = len(results)
b_results = len(list(filter(lambda x: x.winner == simulation.Winner.Engine_B, results))) / len(results) * 100 a_wins = len(list(filter(lambda x: x.winner == simulation.Winner.Engine_A, results)))
draws = len(list(filter(lambda x: x.winner == simulation.Winner.Draw, results))) / len(results) * 100 b_wins = len(list(filter(lambda x: x.winner == simulation.Winner.Engine_B, results)))
draws = len(list(filter(lambda x: x.winner == simulation.Winner.Draw, results)))
print(f"Engine {a.get_name()} won {a_results}% of games") print(f"{games_played} games played")
print(f"Engine {b.get_name()} won {b_results}% of games") print(f"Engine {a.get_name()} won {a_wins} games ({a_wins/games_played:.2%})")
print(f"{draws}% of games resulted in a draw") print(f"Engine {b.get_name()} won {b_wins} games ({b_wins/games_played:.2%})")
print(f"{draws} games ({draws/games_played:.2%}) resulted in a draw")
def read_arguments(): def read_arguments():