def test_perfect_villagers() -> None:
"""Correctly play one round of one night werewolf."""
const.NUM_PLAYERS = 10
const.NUM_CENTER = 3
const.RANDOMIZE_ROLES = False
set_roles(
Role.INSOMNIAC,
Role.VILLAGER,
Role.ROBBER,
Role.VILLAGER,
Role.SEER,
Role.MASON,
Role.TROUBLEMAKER,
Role.VILLAGER,
Role.MASON,
Role.HUNTER,
Role.WOLF,
Role.WOLF,
Role.MINION,
)
random.seed()
stat_tracker = one_night.simulate_game(num_games=10)
stat_results = stat_tracker.get_metric_results()
assert stat_results["villager_wins"] == 1
assert stat_results["correctness_lenient_center"] == 1
assert stat_results["wolf_predictions_one"] == 0
assert stat_results["wolf_predictions_all"] == 1
def test_expectimax_tanner() -> None:
"""Correctly play one round of one night werewolf."""
const.NUM_PLAYERS = 9
const.NUM_CENTER = 3
const.EXPECTIMAX_TANNER = True
const.RANDOMIZE_ROLES = False
set_roles(
Role.INSOMNIAC,
Role.HUNTER,
Role.MASON,
Role.MASON,
Role.ROBBER,
Role.SEER,
Role.TANNER,
Role.TROUBLEMAKER,
Role.VILLAGER,
Role.VILLAGER,
Role.VILLAGER,
Role.WOLF,
)
stat_tracker = one_night.simulate_game(num_games=20)
stat_results = stat_tracker.get_metric_results()
write_results(stat_results, "standard/expectimax_tanner.csv")
assert stat_results["tanner_wins"] > 0.8
def test_small_game(small_game_roles: tuple[Role, ...]) -> None:
"""Correctly play one round of one night werewolf."""
random.seed()
stat_tracker = one_night.simulate_game(num_games=10)
stat_results = stat_tracker.get_metric_results()
assert stat_results["villager_wins"] == 1
def test_random_wolf(standard_game_roles: tuple[Role, ...]) -> None:
"""Correctly play one round of one night werewolf."""
stat_tracker = one_night.simulate_game(num_games=1000)
stat_results = stat_tracker.get_metric_results()
write_results(stat_results, "standard/random_wolf.csv")
assert stat_results["villager_wins"] > 0.8
assert stat_results["tanner_wins"] == 0
assert stat_results["werewolf_wins"] < 0.2
def main() -> None:
"""Simulate play_one_night_werewolf."""
if const.REPLAY:
replay.replay_game_from_state()
else:
enable_logging = const.NUM_GAMES < const.MAX_LOG_GAMES
_ = one_night.simulate_game(
num_games=const.NUM_GAMES,
save_replay=const.SAVE_REPLAY,
enable_tqdm=not enable_logging,
enable_logging=enable_logging,
)
def test_reg_wolf(standard_game_roles: tuple[Role, ...]) -> None:
"""Correctly play one round of one night werewolf."""
const.MULTI_STATEMENT = True
const.USE_REG_WOLF = True
stat_tracker = one_night.simulate_game(num_games=1000)
stat_results = stat_tracker.get_metric_results()
write_results(stat_results, "multistatement/reg_wolf.csv")
assert stat_results["villager_wins"] < 0.51
assert stat_results["tanner_wins"] == 0
assert stat_results["werewolf_wins"] > 0.49
def test_expectimax_wolf(standard_game_roles: tuple[Role, ...]) -> None:
"""Correctly play one round of one night werewolf."""
const.USE_REG_WOLF = True
const.EXPECTIMAX_WOLF = True
stat_tracker = one_night.simulate_game(num_games=500)
stat_results = stat_tracker.get_metric_results()
write_results(stat_results, "standard/expectimax_wolf.csv")
assert stat_results["villager_wins"] < 0.6
assert stat_results["tanner_wins"] == 0
assert stat_results["werewolf_wins"] > 0.4
def test() -> None: # experience_dict as param
"""Run play_one_night_werewolf with a specific experience_dict."""
const.RL_WOLF = False
simulate_game(num_games=const.NUM_GAMES, enable_tqdm=True)