class TestBoard(unittest.TestCase):
def setUp(self):
self.basic_game = Game(board=None, rnd=random.Random(1))
def test_constructor(self):
self.assertEqual(self.basic_game.score(), 0)
game = Game(rnd=random.Random(1))
self.assertEqual(game.board(), self.basic_game.board())
def test_smash(self):
# This doesn't comprehensively test smashing because
# the tests for Board mostly did that already.
for direction in DIRECTIONS:
board = Board()
board = board.update((1, 1), 2)
game = Game(rnd=random.Random(1), board=board)
self.assertTrue(game.smash(direction))
self.assertEqual(game.board()[1, 1], 0)
for direction in DIRECTIONS:
board = Board()
board = board.update((0, 0), 2)
game = Game(rnd=random.Random(1), board=board)
if direction in {UP, LEFT}:
self.assertFalse(game.smash(direction))
else:
self.assertTrue(game.smash(direction))
self.assertEqual(game.board()[0, 0], 0)
strategy = None
if args.strategy == "spinny":
strategy = SpinnyStrategy()
elif args.strategy == "random":
strategy = RandomStrategy()
else:
from strategy.nn.nn_strategy import ModelStrategy
strategy = ModelStrategy(args.strategy, verbose_period=5000)
total = 0
for i in range(args.number_of_games):
game = Game()
running = True
while running:
turn_outcome = game.do_turn(
strategy.get_move(game.board(), game.score()))
if args.verbose:
game.pretty_print()
running = (turn_outcome != GAMEOVER)
strategy.notify_outcome(game.board(), game.score())
if not args.summary:
print(game.score())
total += game.score()
if not (i % 25):
print("...", i, "/", args.number_of_games)
if args.summary:
print("Strategy %s had average score %f after %d games" %
(args.strategy,
(total / args.number_of_games), args.number_of_games))
