Added a random selection even for teams

1 files changed, 5 insertions, 3 deletions

diff --git a/src/match_up/utilities.py b/src/match_up/utilities.py
index dd9568c..7d7de08 100644
--- a/src/match_up/utilities.py
+++ b/src/match_up/utilities.py
@@ -2,7 +2,7 @@ from math import floor
 from typing import Tuple
 from random import shuffle
 from pandas import read_csv
-from random import randint, getrandbits
+from random import randint, getrandbits, sample
 from names import get_first_name, get_last_name
 from match_up.data import Player, DisplayPlayer, Team, Game
 from match_up.data import (
@@ -71,7 +71,7 @@ def _create_groups_of_four_players(
     return court_players
 
 
-def _create_games(courts: dict, players: list[Player], levels: int) -> list[Game]:
+def _create_shuffle_games(courts: dict, players: list[Player], levels: int) -> list[Game]:
     court_players = _create_groups_of_four_players(players, levels)
     games = []
     for court, players in zip(courts, court_players):
@@ -88,7 +88,9 @@ def _create_team_games(courts: dict, teams: dict) -> tuple[list[Game], list[Play
     games = []
     selected_players = []
     for court, (_, players) in zip(courts, teams.items()):
-        selected_players.append(players[:3])
+        players = sample(players, PLAYER_PER_COURT)
+        selected_players.append(players)
+        players.sort(key=lambda x : x.level, reverse=True)
         team1 = Team(
             player1=DisplayPlayer(players[0]), player2=DisplayPlayer(players[3])
         )