From 467dc51d2bce7541d7ffb9484db567ebdd2153fa Mon Sep 17 00:00:00 2001 From: Sakimori Date: Tue, 11 Jan 2022 15:11:51 -0500 Subject: [PATCH] added is player on team bool --- team.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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