def test_move_bunnies():
# create state without bunnies or correct hutches (2 needs to be the first
# hutch)
engine = dice_game.DiceGame(num_players=num_players, type="Bunnies")
state = engine.get_state()
state["rollables"] = [2, 2, 3, 4, 5, 6, 7]
state["allowedMoves"] = {
"roll": 0,
"stay": 0,
"reset": 0,
"moveBunny": 1,
"moveHutch": 0
}
# try to move a die into the bunny container
engine.move({"type": "moveBunny", "value": 0})
assert state["bunnies"][0] == 2
assert state["bunnies"][1] == 0
assert state["rollables"][0] == 0
assert state["hutches"][0] == 0
# check that allowed moves are correct
assert state["allowedMoves"] == {
"roll": 1,
"stay": 1,
"reset": 0,
"moveBunny": 1,
"moveHutch": 1
}
def test_do_roll():
engine = dice_game.DiceGame(num_players=num_players, type="Bunnies")
state = engine.get_state()
# set a state with only bunnies, no rollables
state["rollables"] = [0] * dice_game.NUM_DICE
state["bunnies"] = [1] * dice_game.NUM_DICE
state["movables"] = [0] * dice_game.NUM_DICE
state["allowedMoves"] = {
"roll": 1,
"stay": 1,
"reset": 0,
"moveBunny": 1,
"moveHutch": 1
}
engine.move({"type": "roll", "value": 0})
# check if all dice from bunnies were used as rollables
assert state["rollables"].count(0) == 0
assert state["bunnies"] == [0] * dice_game.NUM_DICE
assert state["extraBunnies"] == (dice_game.NUM_DICE //
2) * 10 + dice_game.NUM_DICE % 2
assert state["allowedMoves"]["roll"] == 0
assert state["allowedMoves"]["reset"] == 0
# check if only some dice can be rolled that unrollables stay unchanged
engine = dice_game.DiceGame(num_players=num_players, type="Bunnies")
state = engine.get_state()
state["rollables"] = [0, 0, 1, 1, 0, 1, 0]
state["bunnies"] = [1 - x for x in state["rollables"]]
state["allowedMoves"] = {
"roll": 1,
"stay": 1,
"reset": 0,
"moveBunny": 1,
"moveHutch": 0
}
old_bunnies = copy.deepcopy(state["bunnies"])
engine.move({"type": "roll", "value": 0})
for i in range(dice_game.NUM_DICE):
if state["rollables"][i] == 0:
assert state["bunnies"][i] == old_bunnies[i]
def test_do_stay():
# check stay move: taking over state from previous player and continue
# playing
engine = dice_game.DiceGame(num_players=num_players, type="Bunnies")
state = engine.get_state()
state["rollables"] = [0, 5, 4, 6, 0, 4, 0]
state["bunnies"] = [1, 0, 0, 0, 1, 0, 0]
state["hutches"] = [0, 0, 0, 0, 0, 0, 2]
state["movables"] = [0, 1, 1, 1, 0, 1, 0]
state["allowedMoves"] = {
"roll": 1,
"stay": 1,
"reset": 0,
"moveBunny": 0,
"moveHutch": 0
}
current_player = copy.deepcopy(state["currentPlayer"])
engine.move({"type": "stay", "value": 0})
assert state["currentPlayer"] == (current_player + 1) % num_players
assert state["boardValue"] == 10 * 2
assert state["scores"][current_player] == 10 * 2
assert state["allowedMoves"] == {
"roll": 1,
"stay": 0,
"reset": 1,
"moveBunny": 0,
"moveHutch": 0
}
# new state where it's not allowed to roll again
state["rollables"] = [0, 5, 4, 6, 0, 4, 0]
state["bunnies"] = [1, 0, 0, 0, 1, 0, 0]
state["hutches"] = [0, 0, 0, 0, 0, 0, 2]
state["movables"] = [0, 1, 1, 1, 0, 1, 0]
state["allowedMoves"] = {
"roll": 0,
"stay": 1,
"reset": 0,
"moveBunny": 0,
"moveHutch": 0
}
current_player = copy.deepcopy(state["currentPlayer"])
engine.move({"type": "stay", "value": 0})
assert state["currentPlayer"] == (current_player + 1) % num_players
def test_score():
engine = dice_game.DiceGame(num_players=2, type="Bunnies")
state = engine.get_state() # get the default state
# create new state to check scoring
state["bunnies"] = [1, 1, 1, 2, 0, 0, 0]
state["hutches"] = [0, 0, 0, 0, 2, 3, 4]
state["extraBunnies"] = 43
state["allowedMoves"] = {
"roll": 1,
"stay": 1,
"reset": 0,
"moveBunny": 0,
"moveHutch": 0
}
current_player = copy.deepcopy(state["currentPlayer"])
engine.move({"type": "stay", "value": 0})
assert 0 in state["scores"]
assert state["scores"][current_player] == (
43 + 10 + 3) * 4 # check that player 0 has correct score
assert state["boardValue"] == (43 + 10 + 3) * 4
# check that next player has correct score
assert state["scores"][state["currentPlayer"]] == 0
def test_do_reset():
engine = dice_game.DiceGame(num_players=num_players, type="Bunnies")
state = engine.get_state()
# set a state that was just passed on
state["rollables"] = [0, 5, 4, 6, 0, 4, 0]
state["bunnies"] = [1, 0, 0, 0, 1, 0, 0]
state["hutches"] = [0, 0, 0, 0, 0, 0, 2]
state["movables"] = [0, 1, 1, 1, 0, 1, 0]
state["allowedMoves"] = {
"roll": 1,
"stay": 0,
"reset": 1,
"moveBunny": 0,
"moveHutch": 0
}
current_player = copy.deepcopy(state["currentPlayer"])
engine.move({"type": "reset", "value": 0})
assert state["rollables"].count(0) == 0
assert state["bunnies"] == [0] * dice_game.NUM_DICE
assert state["hutches"] == [0] * dice_game.NUM_DICE
assert state["movables"] == [1] * dice_game.NUM_DICE
assert state["currentPlayer"] == current_player
assert state["extraBunnies"] == 0
assert state["boardValue"] == 0