def test_add_card(self):
"""Test add_card."""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
player_to_add_to = "Sarah"
card_to_add = "Mr. Green"
game.not_it_but_not_sure_who = [card_to_add]
game.players[player_to_add_to].has.append("Horseshoe")
# Set up at_least_ones. With the addition of Mr. Green, the first entry should be eliminated, and the others should remain
game.players[player_to_add_to].at_least_one = [[
card_to_add, "Horseshoe"
], [card_to_add, "Lead Pipe"], ["Col. Mustard", "Billiard Room"]]
# Add the card
game.add_card(player_to_add_to, card_to_add)
# Make sure card was added to player's list
self.assertIn(card_to_add, game.players[player_to_add_to].has)
# Make sure detective notebook was updated
self.assertEqual(
player_to_add_to, game.detective_notebook[
clueLogic.get_card_type_key(card_to_add)][card_to_add])
# Make sure card was added to the other players' does_not_have list
for player in self.other_players:
if player == player_to_add_to:
continue
self.assertIn(card_to_add, game.players[player].does_not_have)
# Make sure the card was removed from not_it_but_not_sure_who
self.assertEqual([], game.not_it_but_not_sure_who)
# Make sure the player's at_least_one was updated
self.assertEqual(
[[card_to_add, "Lead Pipe"], ["Col. Mustard", "Billiard Room"]],
game.players[player_to_add_to].at_least_one)
# Test behavior when adding the last unknown card to a player
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
player_to_add_to = "Sarah"
card_to_add = "Mr. Green"
# We already know all of Sarah's cards except one
game.players[player_to_add_to].has = [
"M. Brunette", "Horseshoe", "Poison", "Billiard Room", "Fountain",
"Courtyard"
]
# Add the card
game.add_card(player_to_add_to, card_to_add)
# Make sure card was added to player's list
self.assertIn(card_to_add, game.players[player_to_add_to].has)
# Make sure all of the other cards are now in Sarah's does_not_have
for card in self.sample_game["Me"] + self.sample_game[
"Andy"] + self.sample_game["Susan"]:
self.assertIn(card, game.players[player_to_add_to].does_not_have)
def test_conflicting_info(self):
"""Test that nothing terrible happens when conflicting info is entered."""
# Add one of my cards to another player
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
game.add_card("Sarah", "Mme Rose")
self.assertIn("Mme Rose", game.players["Me"].has)
self.assertNotIn("Mme Rose", game.players["Sarah"].has)
# Add a card when we think the player doesn't have that card
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
bad_card = "Horseshoe"
game.enter_not_has("Sarah", bad_card)
game.add_card("Sarah", bad_card)
self.assertIn(bad_card, game.players["Sarah"].has)
self.assertNotIn(bad_card, game.players["Sarah"].does_not_have)
# Add a card when we think another player has that card
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
bad_card = "Horseshoe"
game.add_card("Susan", bad_card)
game.add_card("Sarah", bad_card)
self.assertIn(bad_card, game.players["Sarah"].has)
self.assertNotIn(bad_card, game.players["Susan"].has)
self.assertIn(bad_card, game.players["Susan"].does_not_have)
# Add a card when we think it's in the actual solution
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
bad_card = "Horseshoe"
game.actual_solution.append(bad_card)
game.add_card("Sarah", bad_card)
self.assertIn(bad_card, game.players["Sarah"].has)
self.assertNotIn(bad_card, game.actual_solution)
# Indicate that a player doesn't have a card when we think they have it
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
bad_card = "Horseshoe"
game.add_card("Susan", bad_card)
game.enter_not_has("Susan", bad_card)
self.assertIn(bad_card, game.players["Susan"].does_not_have)
self.assertNotIn(bad_card, game.players["Susan"].has)
def test_update_detective_notebook(self):
"""Test update_detective_notebook"""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
game.actual_solution = ["Sgt. Gray"]
game.not_it_but_not_sure_who = ["Lead Pipe"]
game.players["Sarah"].has = ["Billiard Room"]
game.players["Susan"].has = ["Mrs. Peacock"]
# Initialize detective notebook
expected_detective_notebook = {
"suspects":
{suspect: ""
for suspect in clueLogic.cards["suspects"]},
"weapons": {weapon: ""
for weapon in clueLogic.cards["weapons"]},
"rooms": {room: ""
for room in clueLogic.cards["rooms"]}
}
for suspect in self.my_suspects:
expected_detective_notebook["suspects"][suspect] = "Me"
for weapon in self.my_weapons:
expected_detective_notebook["weapons"][weapon] = "Me"
for room in self.my_rooms:
expected_detective_notebook["rooms"][room] = "Me"
expected_detective_notebook["suspects"]["Sgt. Gray"] = "SOLUTION"
expected_detective_notebook["weapons"]["Lead Pipe"] = "NO"
expected_detective_notebook["suspects"]["Mrs. Peacock"] = "Susan"
expected_detective_notebook["rooms"]["Billiard Room"] = "Sarah"
# Update detective notebook with current information
game.update_detective_notebook()
self.assertDictEqual(expected_detective_notebook,
game.detective_notebook)
def test_narrow_down_guess(self):
"""Test narrow_down_guess."""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
# Test that we get empty lists if the input guess is empty
self.assertEqual(([], []), game.narrow_down_guess([], []))
# If there's exactly one disprover and one card remaining in the guess, we should
# get empty lists returned, and the card should be added to the disprover
guess = ["Mr. Green"]
disprovers = ["Sarah"]
self.assertEqual(([], []), game.narrow_down_guess(guess, disprovers))
self.assertIn(guess[0], game.players[disprovers[0]].has)
# Test that when we have no information to narrow down the guess, the function
# just returns the input
guess = ["Mrs. Peacock", "Lead Pipe", "Fountain"]
disprovers = ["Susan", "Andy", "Sarah"]
self.assertEqual((guess, disprovers),
game.narrow_down_guess(guess, disprovers))
# Test that the function successfully narrows down the possibilities to figure out who
# at least one disprover is. We'll determine that Susan has Mrs. Peacock, and the method
# should eliminate Susan and Mrs. Peacock from the returned lists.
game.players["Susan"].does_not_have.append("Lead Pipe")
game.players["Susan"].does_not_have.append("Fountain")
guess = ["Mrs. Peacock", "Lead Pipe", "Fountain"]
disprovers = ["Susan", "Andy", "Sarah"]
self.assertEqual((["Lead Pipe", "Fountain"], ["Andy", "Sarah"]),
game.narrow_down_guess(guess, disprovers))
self.assertIn("Mrs. Peacock", game.players["Susan"].has)
def test_enter_disproval_of_my_guess(self):
"""Test enter_disproval_of_my_guess."""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
# Test that has gets updated for all disprovers
guessed_suspect = "Mrs. Peacock"
guessed_weapon = "Lead Pipe"
guessed_room = "Fountain"
disprover_suspect = "Susan"
disprover_weapon = "Andy"
disprover_room = "Sarah"
game.enter_disproval_of_my_guess(guessed_suspect, guessed_weapon,
guessed_room, disprover_suspect,
disprover_weapon, disprover_room)
self.assertIn(guessed_suspect, game.players[disprover_suspect].has)
self.assertIn(guessed_weapon, game.players[disprover_weapon].has)
self.assertIn(guessed_room, game.players[disprover_room].has)
self.assertEqual([], game.players[disprover_room].at_least_one)
# Test that all non-disprovers get cards added to does_not_have
# Use the real solution
guessed_suspect = "Sgt. Gray"
guessed_weapon = "Candlestick"
guessed_room = "Studio"
disprover_suspect = None
disprover_weapon = None
disprover_room = None
game.enter_disproval_of_my_guess(guessed_suspect, guessed_weapon,
guessed_room, disprover_suspect,
disprover_weapon, disprover_room)
for player in game.players:
self.assertIn(guessed_suspect, game.players[player].does_not_have)
def test_enter_at_least_one(self):
"""Test enter_at_least_one."""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
game.actual_solution = ["Sgt. Gray", "Studio"]
player_to_enter = "Sarah"
game.players[player_to_enter].has.append("Horseshoe")
game.players[player_to_enter].does_not_have.append("Dining Room")
# If we already know this player has all the cards in the guess, method does nothing
game.enter_at_least_one(player_to_enter, ["Sgt. Gray", "Horseshoe"])
self.assertEqual([], game.players[player_to_enter].at_least_one)
# Test removing cards we know the player doesn't have from the guess and adding a card to has
# if there is only one remaining. We know she doesn't have Sgt. Gray or the Dining Room,
# so she must have the Poison.
game.enter_at_least_one(player_to_enter,
["Sgt. Gray", "Poison", "Dining Room"])
self.assertEqual([], game.players[player_to_enter].at_least_one)
self.assertIn("Poison", game.players[player_to_enter].has)
# Test actually adding to at_least_one when we can't determine any further info
# Sgt. Gray should be weeded out, but we don't know about Rope and Fountain
game.enter_at_least_one(player_to_enter,
["Sgt. Gray", "Rope", "Fountain"])
self.assertIn(["Rope", "Fountain"],
game.players[player_to_enter].at_least_one)
def test_setup_game(self):
"""Test that calling setup_game correctly populates player cards and detective notebook."""
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
self.assertEqual(4, len(game.players))
# Make sure players are properly initialized
self.assertCountEqual(list(self.sample_game.keys()),
list(game.players.keys()))
# The confusingly-named assertCountEqual does an unordered list comparison
self.assertCountEqual(self.sample_game["Me"], game.players["Me"].has)
for player in self.other_players:
self.assertEqual([], game.players[player].has)
self.assertEqual(len(self.sample_game[player]),
game.players[player].num_cards)
self.assertCountEqual(self.sample_game["Me"],
game.players[player].does_not_have)
# Check detective notebook
expected_detective_notebook = {
"suspects":
{suspect: ""
for suspect in clueLogic.cards["suspects"]},
"weapons": {weapon: ""
for weapon in clueLogic.cards["weapons"]},
"rooms": {room: ""
for room in clueLogic.cards["rooms"]}
}
for suspect in self.my_suspects:
expected_detective_notebook["suspects"][suspect] = "Me"
for weapon in self.my_weapons:
expected_detective_notebook["weapons"][weapon] = "Me"
for room in self.my_rooms:
expected_detective_notebook["rooms"][room] = "Me"
self.assertDictEqual(expected_detective_notebook,
game.detective_notebook)
def startGame():
"""Initialize a new game."""
session["game"] = clueLogic.game()
return render_template(
"StartGame.html",
suspects=clueLogic.cards["suspects"],
weapons=clueLogic.cards["weapons"],
rooms=clueLogic.cards["rooms"]
)
def __init__(self, seed=None):
# Specify a random seed so the game is reproducible
if not seed:
seed = random.randrange(sys.maxsize)
random.seed(seed)
print("Random seed was:", seed)
# Create deck of cards and shuffle them
all_cards = clueLogic.cards["suspects"] + clueLogic.cards[
"weapons"] + clueLogic.cards["rooms"]
random.shuffle(all_cards)
# Extract actual solution
actual_suspect, actual_weapon, actual_room = self.generate_random_guess(
)
self.actual_solution = [actual_suspect, actual_weapon, actual_room]
print("Actual solution: ", self.actual_solution)
for sol in self.actual_solution:
all_cards.remove(sol)
# Initialize the players and their hands
num_players = random.randint(2, 10)
print(f"Number of players: {num_players}")
hands = [all_cards[i::num_players] for i in range(0, num_players)]
random.shuffle(hands)
self.players = {}
# Assign a hand to each other player
for i in range(0, num_players - 1):
player_cards = random.choice(hands)
self.players[f"Player_{i + 1}"] = player_cards
hands.remove(player_cards)
# The reamining hand is mine
my_cards = hands[0]
for player in self.players:
print(player, self.players[player])
print("Me", my_cards)
# Initialize game
my_suspects = []
my_weapons = []
my_rooms = []
for card in my_cards:
card_type = clueLogic.get_card_type_key(card)
if card_type == "suspect":
my_suspects.append(card)
elif card_type == "weapons":
my_weapons.append(card)
else:
my_rooms.append(card)
other_players_init = []
for player in self.players:
other_players_init.append([player, len(self.players[player])])
self.game = clueLogic.game()
self.game.setup_game(my_suspects, my_weapons, my_rooms,
other_players_init)
def test_enter_snoop(self):
"""Test enter_snoop."""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
snooped_player = "Sarah"
snooped_card = "Mr. Green"
# Snoop
game.enter_snoop(snooped_player, snooped_card)
self.assertIn(snooped_card, game.players[snooped_player].has)
def test_add_to_actual_solution(self):
"""Test add_to_actual_solution."""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
# Add to the actual solution
card_to_add = "Sgt. Gray"
game.add_to_actual_solution(card_to_add)
# Check that the card was added and that all other cards of this type are in not_it_but_not_sure_who
self.assertIn(card_to_add, game.actual_solution)
for card in clueLogic.cards["suspects"]:
if card in self.sample_game["Me"] or card == card_to_add:
self.assertNotIn(card, game.not_it_but_not_sure_who)
else:
self.assertIn(card, game.not_it_but_not_sure_who)
def replayFromLog():
log = [l for l in session["game"].log]
log.reverse()
# Get the bad entries from the form
bad_actions = request.form.getlist("actions")
for action in bad_actions:
if action in log:
log.remove(action)
# Reinitialize a game and play through it automatically from scratch
# using the cleaned-up log
session["game"] = clueLogic.game()
session["game"].play_game_from_log_text(log)
# Render notebook
return renderMyNotebook()
def test_check_for_solution_by_elimination(self):
"""Test check_for_solution_by_elimination"""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
# Initialize detective notebook so that there is only one remaining suspect
game.add_card("Susan", "Mrs. Peacock")
game.add_card("Susan", "Col. Mustard")
game.add_card("Susan", "Mrs. White")
game.add_card("Susan", "Miss Peach")
game.add_card("Andy", "Miss Scarlet")
game.add_card("Sarah", "M. Brunette")
game.add_card("Sarah", "Mr. Green")
# Run the method and make sure the the remaining Sgt. Gray is tagged with SOLUTION
game.check_for_solution_by_elimination()
expected_detective_notebook = {
"suspects":
{suspect: ""
for suspect in clueLogic.cards["suspects"]},
"weapons": {weapon: ""
for weapon in clueLogic.cards["weapons"]},
"rooms": {room: ""
for room in clueLogic.cards["rooms"]}
}
for suspect in self.my_suspects:
expected_detective_notebook["suspects"][suspect] = "Me"
for weapon in self.my_weapons:
expected_detective_notebook["weapons"][weapon] = "Me"
for room in self.my_rooms:
expected_detective_notebook["rooms"][room] = "Me"
expected_detective_notebook["suspects"]["Mrs. Peacock"] = "Susan"
expected_detective_notebook["suspects"]["Col. Mustard"] = "Susan"
expected_detective_notebook["suspects"]["Mrs. White"] = "Susan"
expected_detective_notebook["suspects"]["Miss Peach"] = "Susan"
expected_detective_notebook["suspects"]["Miss Scarlet"] = "Andy"
expected_detective_notebook["suspects"]["M. Brunette"] = "Sarah"
expected_detective_notebook["suspects"]["Mr. Green"] = "Sarah"
expected_detective_notebook["suspects"]["Sgt. Gray"] = "SOLUTION"
self.assertDictEqual(expected_detective_notebook,
game.detective_notebook)
# Make sure it was added to the actual solution
self.assertEqual(["Sgt. Gray"], game.actual_solution)
# Make sure card was added to everyone's does_not_have list
for player in self.other_players:
self.assertIn("Sgt. Gray", game.players[player].does_not_have)
def test_enter_other_players_guess(self):
"""Test enter_other_players_guess."""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
# Basic test 1: adding to a disprover's at_least_one
guesser = "Susan"
guessed_suspect = "Prof. Plum" # I have this one
guessed_weapon = "Wrench" # Andy has this one
guessed_room = "Library" # Susan is being tricky and has this one herself, but we don't know it.
disprovers = ["Andy"] # Sarah could not disprove.
game.enter_other_players_guess(guesser, guessed_suspect,
guessed_weapon, guessed_room,
disprovers)
# Sarah is not a disprover, so her does_not_have gets updated to include all cards in the guess
self.assertIn(guessed_weapon, game.players["Sarah"].does_not_have)
self.assertIn(guessed_room, game.players["Sarah"].does_not_have)
# Andy gets the guess items added to at_leaest_one
self.assertIn([guessed_weapon, guessed_room],
game.players["Andy"].at_least_one)
# not_it_but_not_sure_who should not be updated
self.assertEqual([], game.not_it_but_not_sure_who)
# Basic test 2: special case for not_it_but_not_sure_who
guesser = "Sarah"
guessed_suspect = "Prof. Plum" # I have this one
guessed_weapon = "Rope" # Andy has this one
guessed_room = "Trophy Room" # Susan has this one
disprovers = ["Susan", "Andy"] # Andy, Susan, and I all disproved
game.enter_other_players_guess(guesser, guessed_suspect,
guessed_weapon, guessed_room,
disprovers)
# We should be able to determine that Sarah has none of these cards
self.assertIn(guessed_weapon, game.players["Sarah"].does_not_have)
self.assertIn(guessed_room, game.players["Sarah"].does_not_have)
# Andy and Susan get the guess items added to at_leaest_one
self.assertIn([guessed_weapon, guessed_room],
game.players["Andy"].at_least_one)
self.assertIn([guessed_weapon, guessed_room],
game.players["Susan"].at_least_one)
# not_it_but_not_sure_who should be updated
self.assertCountEqual([guessed_weapon, guessed_room],
game.not_it_but_not_sure_who)
def test_remove_known_from_guess(self):
"""Test remove_known_from_guess."""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
game.players["Susan"].has.append("Mrs. Peacock")
game.actual_solution = ["Sgt. Gray", "Studio"]
# Function should remove something that the guesser has
guess = ["Mrs. Peacock", "Lead Pipe", "Fountain"]
self.assertEqual(["Lead Pipe", "Fountain"],
game.remove_known_from_guess("Susan", guess))
# Function should remove one of my cards from the guess
guess = ["Mrs. Peacock", "Revolver", "Fountain"]
self.assertEqual(["Mrs. Peacock", "Fountain"],
game.remove_known_from_guess("Andy", guess))
# Function should remove known solution cards from guess
guess = ["Sgt. Gray", "Lead Pipe", "Studio"]
self.assertEqual(["Lead Pipe"],
game.remove_known_from_guess("Andy", guess))
def test_enter_not_has(self):
"""Test enter_not_has."""
# Set up a contrived game with circumstances useful for testing
game = clueLogic.game()
game.setup_game(self.my_suspects, self.my_weapons, self.my_rooms,
self.other_players_init)
player_to_enter = "Sarah"
card_to_enter = "Sgt. Gray"
game.players[player_to_enter].has.append("Horseshoe")
for player in game.players:
if player not in ["Me", player_to_enter]:
game.players[player].does_not_have.append(card_to_enter)
# Set up at_least_ones. With the removal of Miss Scarlet, the first entry should be eliminated and the Poison added to has
# The second entry should get Miss Scarlet removed but leave the rest.
game.players[player_to_enter].at_least_one = [[
card_to_enter, "Poison"
], [card_to_enter, "Horseshoe", "Billiard Room"]]
# Add the card
game.enter_not_has(player_to_enter, card_to_enter)
# Make sure card was added to player's does_not_have list
self.assertIn(card_to_enter,
game.players[player_to_enter].does_not_have)
# Make sure the player's at_least_one was updated
self.assertEqual([["Horseshoe", "Billiard Room"]],
game.players[player_to_enter].at_least_one)
# Make sure actual_solution was updated
self.assertEqual([card_to_enter], game.actual_solution)
# Make sure detective notebook was updated
self.assertEqual("SOLUTION",
game.detective_notebook["suspects"][card_to_enter])
for card in clueLogic.cards["suspects"]:
if card in self.my_suspects:
self.assertEqual("Me",
game.detective_notebook["suspects"][card])
elif card != card_to_enter:
self.assertEqual("NO",
game.detective_notebook["suspects"][card])
import sys
sys.path.insert(1, r'E:\Code\Clue_app_Flask')
import clueLogic
log_file = r"E:\Code\Clue_app_Flask\tests\test_game_log.txt"
with open(log_file, "r") as f:
log = f.read().splitlines()
game = clueLogic.game()
game.play_game_from_log_text(log)
