def test_add_move():
state1 = SploofState("""\
A C E G
7 R R R R 7
5 . . . . 5
3 R W B R 3
1 R R R R 1
A C E G
B D F
6 . . . 6
4 . . . 4
2 W B . 2
B D F
>W(2,2)
""")
expected_display = """\
A C E G
7 R R R R 7
5 W . . . 5
3 R W B R 3
1 R R R R 1
A C E G
B D F
6 . . . 6
4 . . . 4
2 W B . 2
B D F
>B(2,1)
"""
state2 = state1.make_move(8)
display = state2.display()
active_player = state2.get_active_player()
assert display == expected_display
assert active_player == state2.BLACK
def test_let_opponent_win():
state1 = SploofState("""\
A C E G
7 . . W W 7
5 . . W W 5
3 . . W W 3
1 B . R B 1
A C E G
B D F
6 . . W 6
4 . . . 4
2 . . W 2
B D F
>B(16,0)
""")
state2 = state1.make_move(32)
assert state2.get_winner() == state2.WHITE