-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_engine_vs_engine.py
136 lines (124 loc) · 5.01 KB
/
test_engine_vs_engine.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
from engine import EngineInterface as EngineInterface1
from test_engine import EngineInterface as EngineInterface2
from multiprocessing import Pool
import time
engine_interface_1 = EngineInterface1(3)
engine_interface_1.name = "engine"
engine_interface_2 = EngineInterface2(3)
engine_interface_2.name = "test_engine"
def single_game_pool_wrapper(argument_container):
(engine1, engine2, engine1_begin) = argument_container
return single_game(engine1, engine2, engine1_begin)
def single_game(engine1, engine2, engine1_begin, print_move_times=False):
"""engine1 and engine2 are instances of EngineInterface.
Return (result, engine1 time, engine2 time)
result is 0 if the game is a draw, 1 if engine1 win and 2 if engine2 win.
"""
if engine1_begin:
return game(engine1, engine2, print_move_times)
else:
(result, engine2_time, engine1_time) = game(engine2, engine1, print_move_times)
if result == 0:
return (0, engine1_time, engine2_time)
if result == 1:
return (2, engine1_time, engine2_time)
if result == 2:
return (1, engine1_time, engine2_time)
def game(engine1, engine2, print_move_times=False):
"""engine1 and engine2 are instances of EngineInterface.
engine1 makes the first move.
Return (result, engine1 time, engine2 time)
result is 0 if the game is a draw, 1 if engine1 win and 2 if engine2 win.
"""
total_time_engine_1 = 0
total_time_engine_2 = 0
engine1.new_game()
engine2.new_game()
while True:
t0 = time.perf_counter()
column_number = engine1.engine_move()
t1 = time.perf_counter()
total_time_engine_1 += t1 - t0
if print_move_times:
print(engine1.name + ": Move time:", t1 - t0, "s")
engine1.make_move(column_number)
engine2.make_move(column_number)
if engine1.four_in_a_row():
return (1, total_time_engine_1, total_time_engine_2)
if engine1.draw():
return (0, total_time_engine_1, total_time_engine_2)
t0 = time.perf_counter()
column_number = engine2.engine_move()
t1 = time.perf_counter()
total_time_engine_2 += t1 - t0
if print_move_times:
print(engine2.name + ": Move time:", t1 - t0, "s")
engine1.make_move(column_number)
engine2.make_move(column_number)
if engine2.four_in_a_row():
return (2, total_time_engine_1, total_time_engine_2)
if engine2.draw():
return (0, total_time_engine_1, total_time_engine_2)
def games(engine1, engine2, number_of_games, print_move_times=False):
"""Let engine1 and engine2 play several games against each other.
Each begin every second game."""
engine1_wins = 0
engine2_wins = 0
engine1_time = 0
engine2_time = 0
draws = 0
for n in range(1, number_of_games + 1):
print("Game", n)
engine1_begin = n % 2 == 0
(result, t1, t2) = single_game(engine1, engine2, engine1_begin, print_move_times)
print(engine1.name + " total time:", t1)
print(engine2.name + " total time:", t2)
if result == 1:
print(engine1.name + " wins" + "\n")
engine1_wins += 1
elif result == 2:
print(engine2.name + " wins" + "\n")
engine2_wins += 1
else:
print("Draw\n")
draws += 1
engine1_time += t1
engine2_time += t2
print(engine1.name + " wins: " + str(engine1_wins) +
"\n" + engine2.name + " wins: " + str(engine2_wins) + "\ndraws: " + str(draws))
print(engine1.name + " total time:", engine1_time, "s")
print(engine2.name + " total time:", engine2_time, "s")
def games_multiprocessing(engine1, engine2, number_of_games):
engine1_wins = 0
engine2_wins = 0
engine1_time = 0
engine2_time = 0
draws = 0
for n in range(number_of_games // 40):
with Pool() as p:
output = p.map(single_game_pool_wrapper,
[(engine1, engine2, True), (engine1, engine2, False)]*20)
i = 1
for (result, t1, t2) in output:
print("Game", n*40 + i)
i += 1
print(engine1.name + " total time:", t1)
print(engine2.name + " total time:", t2)
if result == 1:
print(engine1.name + " wins" + "\n")
engine1_wins += 1
elif result == 2:
print(engine2.name + " wins" + "\n")
engine2_wins += 1
else:
print("Draw\n")
draws += 1
engine1_time += t1
engine2_time += t2
# Print a sum of the results.
print(engine1.name + " wins: " + str(engine1_wins) +
"\n" + engine2.name + " wins: " + str(engine2_wins) + "\ndraws: " + str(draws))
print(engine1.name + " total time:", engine1_time, "s")
print(engine2.name + " total time:", engine2_time, "s")
#games(engine_interface_1, engine_interface_2, 40, False)
games_multiprocessing(engine_interface_1, engine_interface_2, 160*1)