diff --git a/team.py b/team.py index ff1f31c..9dbbfcd 100644 --- a/team.py +++ b/team.py @@ -1,4 +1,4 @@ -import player +from player import Skater, Goalie, Player from random import sample class Team(object): @@ -14,5 +14,9 @@ class Team(object): self.name = name self.shortname = shortname + def isPlayerOnTeam(self, player:Player): + return player in self.roster or player in self.goalies + + def chooseGoalie(self): return sample(self.goalies,1) \ No newline at end of file