forked from suragnair/alpha-zero-general
-
Notifications
You must be signed in to change notification settings - Fork 0
/
BoardTest.py
53 lines (44 loc) · 1.32 KB
/
BoardTest.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
from cubicup.CubicupGame import CubicupGame as Game
import numpy as np
from multiprocessing import Pool
from functools import partial
def play_game(game,i):
player = -1
board = game.getInitBoard()
while 0 == game.getGameEnded(board, player):
moves = game.getValidMoves(board, player)
if moves.any():
posible = np.where(moves == 1)[0]
rand_choice = np.random.randint(len(posible))
board, player = game.getNextState(board, player, posible[rand_choice])
else:
player = -player
return game.getGameDraw(board)*1
N = 100000
nums = np.zeros(N)
runs = np.zeros(N)
n = 6
counter = 0
stones = np.zeros((n*(n+1)*(n+2)/6, ))
#
# b = np.zeros((n,) * 3, dtype=int)
# for i in range(n):
# for a in range(n):
# for b in range(n):
# for c in range(n):
# if a + b + c == i:
# print (a, b, c)
# print counter
# counter +=1
# for i in range(N):
game = Game(n)
pool = Pool(processes=4)
func = partial(play_game, game)
for i in pool.imap(func, range(N)):
print i
# print board
# nums[i] = game.getScore(board, 1)
# if runs[i] != 1 : print(board[game.supported_dict[0]] , " _ " , sum(board[game.supported_dict[0]]) ," : ", runs[i])
print 'done'
#print sum(nums)
print sum(runs)