Bot recognizes a draw

Author: kerzhner

bot/player.py

@@ -48,11 +48,21 @@ def _transfer_stake(hex_message, multiplier=1):
     return coder.increment_state_balance(hex_message, 1, multiplier * stake)
 
 
-def stake_move(hex_message):
+def check_end_of_game(hex_message):
+    if strategy.check_win(hex_message):
+        hex_message = coder.update_game_position(hex_message, 3)
+        return transfer_double_stake(hex_message)
+    if strategy.check_draw(hex_message):
+        hex_message = coder.update_game_position(hex_message, 4)
+        return transfer_stake(hex_message)
+    return transfer_double_stake(hex_message)
+
+
+def transfer_double_stake(hex_message):
     return _transfer_stake(hex_message, 2)
 
 
-def stake_first_move(hex_message):
+def transfer_stake(hex_message):
     return _transfer_stake(hex_message)
 
 
@@ -67,19 +77,18 @@ def play_cross_move(hex_message):
     crosses = coder.get_game_crosses(hex_message) + move
     return coder.update_crosses(hex_message, crosses)
 
-# Need to fix stake
-
 
 def from_resting(_hex_message):
-    return [coder.update_noughts, stake_first_move, play_cross_move, coder.change_game_position]
+    return [coder.update_noughts, transfer_stake, play_cross_move, coder.change_game_position]
 
 
 def from_oplay(_hex_message):
-    return [stake_move, play_cross_move, lambda h_message: coder.change_game_position(h_message, -1)]
+    return [play_cross_move,
+            lambda h_message: coder.change_game_position(h_message, -1), check_end_of_game]
 
 
 def from_xplay(_hex_message):
-    return [stake_move, play_nought_move, coder.change_game_position]
+    return [play_nought_move, coder.change_game_position, check_end_of_game]
 
 
 def from_victory(hex_message):
@@ -89,12 +98,16 @@ def from_victory(hex_message):
     return [coder.new_game]
 
 
+def from_draw(_hex_message):
+    return [coder.new_game]
+
+
 GAME_STATES = (
     from_resting,
     from_xplay,
     from_oplay,
     from_victory,
-    lambda x: undefined_game_position('FromDraw'),
+    from_draw,
 )
 
 

bot/strategy.py

@@ -7,6 +7,9 @@
     64, 128, 256,
 ]
 
+WINNING_POSITIONS = [448, 56, 7, 292, 146, 73, 273, 84]
+DRAW = 511
+
 
 def next_move(hex_message):
     noughts = get_game_noughts(hex_message)
@@ -16,3 +19,19 @@ def next_move(hex_message):
         move = VALID_MOVES[random.randint(0, 8)]
         if noughts | move != noughts and crosses | move != crosses:
             return move
+
+
+def check_win(hex_message):
+    noughts = get_game_noughts(hex_message)
+    crosses = get_game_crosses(hex_message)
+    if noughts in WINNING_POSITIONS or crosses in WINNING_POSITIONS:
+        return True
+    return False
+
+
+def check_draw(hex_message):
+    noughts = get_game_noughts(hex_message)
+    crosses = get_game_crosses(hex_message)
+    if noughts + crosses == DRAW:
+        return True
+    return False

test/test_app.py

@@ -106,9 +106,33 @@ def test_black_box_from_victory(test_app):
         state_machine_transition(state, new_state)
 
 
+def test_black_box_from_draw(test_app, mocker):
+    with test_app.test_request_context('channel_message'):
+        mocker.patch('random.randint', lambda _x, _y: 2)
+        state = '0000000000000000000000006251cac0089a95826b375f9e4d8638b8d472eee400000000000000000000000000000000000000000000000000000000000000050000000000000000000000000000000000000000000000000000000000000002000000000000000000000000A317E78f9F62Bc4959D0682F30043B9C0797C74F00000000000000000000000055de2e479f3d0183d95f5a969beeb3a147f600490000000000000000000000000000000000000000000000000000000000000002000000000000000000000000000000000000000000000000000000000000000c00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000011c37937e080000000000000000000000000000000000000000000000000000011c37937e08000000000000000000000000000000000000000000000000000000000000000000400000000000000000000000000000000000000000000000000038d7ea4c68000000000000000000000000000000000000000000000000000000000000000009c0000000000000000000000000000000000000000000000000000000000000163'
+        new_state = '0000000000000000000000006251cac0089a95826b375f9e4d8638b8d472eee400000000000000000000000000000000000000000000000000000000000000050000000000000000000000000000000000000000000000000000000000000002000000000000000000000000A317E78f9F62Bc4959D0682F30043B9C0797C74F00000000000000000000000055de2e479f3d0183d95f5a969beeb3a147f600490000000000000000000000000000000000000000000000000000000000000002000000000000000000000000000000000000000000000000000000000000000d00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000011c37937e080000000000000000000000000000000000000000000000000000011c37937e08000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000038d7ea4c68000'
+        state_machine_transition(state, new_state)
+
+
 def test_black_box_from_resting(test_app, mocker):
     with test_app.test_request_context('channel_message'):
         mocker.patch('random.randint', lambda _x, _y: 0)
         state = '00000000000000000000000007f96aa816c1f244cbc6ef114bb2b023ba54a2eb000000000000000000000000000000000000000000000000000000000000000500000000000000000000000000000000000000000000000000000000000000020000000000000000000000001251659F5A4897bBebeB949E6Df22697d43C42dC000000000000000000000000A317E78f9F62Bc4959D0682F30043B9C0797C74F0000000000000000000000000000000000000000000000000000000000000002000000000000000000000000000000000000000000000000000000000000000a000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000044364c5bb00000000000000000000000000000000000000000000000000000002d79883d2000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000b5e620f48000'
         new_state = '00000000000000000000000007f96aa816c1f244cbc6ef114bb2b023ba54a2eb000000000000000000000000000000000000000000000000000000000000000500000000000000000000000000000000000000000000000000000000000000020000000000000000000000001251659F5A4897bBebeB949E6Df22697d43C42dC000000000000000000000000A317E78f9F62Bc4959D0682F30043B9C0797C74F0000000000000000000000000000000000000000000000000000000000000002000000000000000000000000000000000000000000000000000000000000000b000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000038d7ea4c6800000000000000000000000000000000000000000000000000000038d7ea4c6800000000000000000000000000000000000000000000000000000000000000000010000000000000000000000000000000000000000000000000000b5e620f4800000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000001'
         state_machine_transition(state, new_state)
+
+
+def test_black_box_to_bot_win(test_app, mocker):
+    with test_app.test_request_context('channel_message'):
+        mocker.patch('random.randint', lambda _x, _y: 2)
+        state = '00000000000000000000000007f96aa816c1f244cbc6ef114bb2b023ba54a2eb000000000000000000000000000000000000000000000000000000000000000500000000000000000000000000000000000000000000000000000000000000020000000000000000000000001251659F5A4897bBebeB949E6Df22697d43C42dC000000000000000000000000A317E78f9F62Bc4959D0682F30043B9C0797C74F0000000000000000000000000000000000000000000000000000000000000002000000000000000000000000000000000000000000000000000000000000000700000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000002d79883d2000000000000000000000000000000000000000000000000000000044364c5bb000000000000000000000000000000000000000000000000000000000000000000020000000000000000000000000000000000000000000000000000b5e620f4800000000000000000000000000000000000000000000000000000000000000000180000000000000000000000000000000000000000000000000000000000000120'
+        new_state = '00000000000000000000000007f96aa816c1f244cbc6ef114bb2b023ba54a2eb000000000000000000000000000000000000000000000000000000000000000500000000000000000000000000000000000000000000000000000000000000020000000000000000000000001251659F5A4897bBebeB949E6Df22697d43C42dC000000000000000000000000A317E78f9F62Bc4959D0682F30043B9C0797C74F00000000000000000000000000000000000000000000000000000000000000020000000000000000000000000000000000000000000000000000000000000008000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000016bcc41e900000000000000000000000000000000000000000000000000000005af3107a4000000000000000000000000000000000000000000000000000000000000000000030000000000000000000000000000000000000000000000000000b5e620f4800000000000000000000000000000000000000000000000000000000000000000180000000000000000000000000000000000000000000000000000000000000124'
+        state_machine_transition(state, new_state)
+
+
+def test_black_box_to_bot_draw(test_app, mocker):
+    with test_app.test_request_context('channel_message'):
+        mocker.patch('random.randint', lambda _x, _y: 1)
+        state = '0000000000000000000000006251cac0089a95826b375f9e4d8638b8d472eee400000000000000000000000000000000000000000000000000000000000000050000000000000000000000000000000000000000000000000000000000000002000000000000000000000000A317E78f9F62Bc4959D0682F30043B9C0797C74F00000000000000000000000055de2e479f3d0183d95f5a969beeb3a147f60049000000000000000000000000000000000000000000000000000000000000000200000000000000000000000000000000000000000000000000000000000000160000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000001550f7dca70000000000000000000000000000000000000000000000000000000e35fa931a0000000000000000000000000000000000000000000000000000000000000000000200000000000000000000000000000000000000000000000000038d7ea4c6800000000000000000000000000000000000000000000000000000000000000000b1000000000000000000000000000000000000000000000000000000000000014c'
+        new_state = '0000000000000000000000006251cac0089a95826b375f9e4d8638b8d472eee400000000000000000000000000000000000000000000000000000000000000050000000000000000000000000000000000000000000000000000000000000002000000000000000000000000A317E78f9F62Bc4959D0682F30043B9C0797C74F00000000000000000000000055de2e479f3d0183d95f5a969beeb3a147f600490000000000000000000000000000000000000000000000000000000000000002000000000000000000000000000000000000000000000000000000000000001700000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000011c37937e080000000000000000000000000000000000000000000000000000011c37937e08000000000000000000000000000000000000000000000000000000000000000000400000000000000000000000000000000000000000000000000038d7ea4c6800000000000000000000000000000000000000000000000000000000000000000b1000000000000000000000000000000000000000000000000000000000000014e'
+        state_machine_transition(state, new_state)