def test_saving_the_winner_of_a_game(save_path):
"""Test the saving and retrieval of game rankings."""
# First, create a completely new statistics object
stats = Statistics(save_path=save_path)
# Save the results of two consecutive games
stats.save(ranking=["gold1", "bronze1", "silver1", "loser1"])
stats.save(ranking=["gold2", "bronze2", "silver2", "loser2"])
# Check if the results are immediatly retrievable
expected_rankings = [
["gold1", "bronze1", "silver1", "loser1"],
["gold2", "bronze2", "silver2", "loser2"],
]
assert stats.data.all_rankings == expected_rankings
# And then check if the same data can be retrieved after a "restart"
del stats
new_stats = Statistics(save_path=save_path)
assert new_stats.data.all_rankings == expected_rankings
class Game:
"""This class represents the whole game."""
def __init__(self) -> None:
"""Initialize the game with all the components."""
pg.init()
pg.font.init()
self.screen: graphics.Screen = graphics.Screen()
# Statistics
self.statistics = Statistics()
# Sliders
self.slides: List[graphics.Slider] = [
graphics.Slider(Slider("Drag coefficient", 50, 0, 100, 1150, 820)),
graphics.Slider(Slider("Friction", 10000, 0, 30000, 1150, 880)),
graphics.Slider(Slider('Player 1 mass', 1, 0.01, 2, 1300, 820)),
graphics.Slider(Slider('Player 2 mass', 1, 0.01, 2, 1300, 880)),
]
# Players
self.players: List[player.Player] = [
player.Player(200,
200,
ps.Player1Settings,
mass=self.slides[2],
phone=ConnectPhone()),
player.Player(300, 300, ps.Player2Settings, mass=self.slides[3]),
]
# Plots
self.graphs: List[graphics.Graph] = [
graphics.Graph(self.players, "v"),
graphics.Graph(self.players, "a", position=(1100, 400))
]
# Buttons
self.buttons: List[graphics.Button] = [
graphics.Button(
Button("Start!", 700, 600, 100, 50, Colors.GREEN.value,
Colors.BLUE.value))
]
# End screen buttons
self.end_btn: List[graphics.Button] = [
graphics.Button(
Button("Retry", 600, 600, 100, 50, Colors.GREEN.value,
Colors.BLUE.value)),
graphics.Button(
Button("Quit", 800, 600, 100, 50, Colors.RED.value,
Colors.BLUE.value)),
]
# Layers
self.layers: List[arena.ArenaLayer] = [
arena.FrictionLayer(np.ones((ArenaSettings.x, ArenaSettings.y)),
self.slides[1].get_value),
arena.AirResistanceLayer(self.slides[0].get_value),
]
# Arena
self.arena: arena.Arena = arena.Arena(ArenaSettings.x,
ArenaSettings.y,
layers=self.layers)
# Physic engine
self.physics: Physics = Physics(self)
# Input boxes
self.input_boxes = [
graphics.InputBox(500, 400, 200, 32, text='player 1 name'),
graphics.InputBox(800, 400, 200, 32, text='player 2 name')
]
# Score
self.score_table = [
graphics.Text(
Texts("Win and draw probabilities", ScreenSettings.width / 2,
200, 115)),
graphics.Text(Texts('', ScreenSettings.width / 2, 260, 80)),
graphics.Text(Texts('', ScreenSettings.width / 2, 320, 80)),
graphics.Text(Texts('', ScreenSettings.width / 2, 380, 80)),
]
def run(self) -> None:
"""Run the whole game loop."""
clock: pg.time.Clock = pg.time.Clock()
while 1:
if self.run_intro():
break
# Set player name
self.players[0].name = self.input_boxes[0].text
self.players[1].name = self.input_boxes[1].text
self.cont_game: bool = True
while self.cont_game:
if not self.run_game(clock):
# Give the winner points
if self.players[0].health_bar.health > self.players[
1].health_bar.health:
self.players[0].score += 1
else:
self.players[1].score += 1
stats = [
player.name
for player in sorted(self.players,
key=lambda x: x.health_bar.health,
reverse=True)
]
self.statistics.save(stats)
stats = self.statistics.win_probability(
[self.players[0].name, self.players[1].name])
# Add highscore
self.score_table[
1].msg = f"{self.players[0].name}: {stats['player_1']:.2f}"
self.score_table[
2].msg = f"{self.players[1].name}: {stats['player_2']:.2f}"
self.score_table[3].msg = f"Draw: {stats['draw']:.2f}"
while 1:
if self.run_end():
self.end_btn[0].clicked = False
break
self.reset_game()
pg.quit()
def run_intro(self) -> bool:
"""Run the game introduction."""
for event in pg.event.get():
if event.type == pg.QUIT:
pg.quit()
quit()
for box in self.input_boxes:
box.handle_event(event)
for box in self.input_boxes:
box.update()
# Fill the screen with white
self.screen.screen.fill(Colors.WHITE.value)
# Draw text onto the screen
graphics.Text(
Texts(
"EiT gruppe rød",
ScreenSettings.width / 2,
300,
115,
)).draw(self.screen.screen)
# Draw the buttions
list(map(lambda x: x.draw(self.screen.screen), self.buttons))
# Input boxes
for box in self.input_boxes:
box.draw(self.screen.screen)
pg.display.update()
return self.buttons[0].clicked
def run_game(self, clock: pg.time.Clock) -> bool:
"""Run the game."""
run: bool = True
# Check for events
for event in pg.event.get():
if event.type == pg.QUIT:
run = False
# Code for slides
elif event.type == pg.MOUSEBUTTONDOWN:
pos = pg.mouse.get_pos()
for s in self.slides:
if s.button_rect.collidepoint(pos):
s.hit = True
elif event.type == pg.MOUSEBUTTONUP:
for s in self.slides:
s.hit = False
# Fill the screen with black to remove old drawings
self.screen.screen.fill(Colors.BLACK.value)
# Draw arena
self.arena.draw(self.screen.screen)
# Move players
self.physics.move_players()
# Draw players
for p in self.players:
p.update_health()
p.draw(self.screen.screen)
# Draw plots
for graph in self.graphs:
graph.draw(self.screen.screen)
# Move sliders
for s in self.slides:
if s.hit:
pass
s.move()
# Draw sliders
for s in self.slides:
s.draw(self.screen.screen)
pg.display.flip()
clock.tick(60)
if True in [p.health_bar.health <= 0 for p in self.players]:
return False
return run
def run_end(self) -> bool:
"""Run when one player dies."""
for event in pg.event.get():
if event.type == pg.QUIT:
pg.quit()
quit()
# Fill the screen with white
self.screen.screen.fill(Colors.WHITE.value)
# Draw text onto the screen
list(map(lambda x: x.draw(self.screen.screen), self.score_table))
# Draw the buttions
list(map(lambda x: x.draw(self.screen.screen), self.end_btn))
if self.end_btn[1].clicked:
self.cont_game = False
pg.display.update()
return self.end_btn[0].clicked or self.end_btn[1].clicked
def get_players(self) -> List[player.Player]:
"""Return all player in game."""
return self.players
def get_arena(self) -> arena.Arena:
"""Return game arena."""
return self.arena
def reset_game(self):
list(map(lambda x: x.reset_position(), self.players))
list(map(lambda x: x.health_bar.reset(), self.players))