class TestGameState(unittest.TestCase):
def setUp(self):
self.gs = GameState()
def test_outputs_to_move_max(self):
outputs = np.linspace(0.0, 1.0, 100)
self.gs.outputs_to_move_max(outputs)
self.assertTrue((self.gs.board == np.array([[-1, -1, -2, -1, -1],
[0, 0, 0, 0, 1],
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 0],
[1, 1, 2, 1, 0]])).all())
self.gs.outputs_to_move_max(outputs)
self.assertFalse((self.gs.board == np.array([[-1, -1, -2, -1, -1],
[0, 0, 0, 0, 1],
[0, 0, 0, 0, 0],
[0, 0, 0, 0, 0],
[1, 1, 2, 1, 0]])).all())
def test_outputs_to_move_random(self):
outputs = np.linspace(0.0, 1.0, 100)
outputs /= np.sum(outputs)
self.gs.outputs_to_move_random(outputs)
def test_flip(self):
self.assertTrue((self.gs.to_inputs() == self.gs.to_inputs(True)).all())
