fixing unit test to reflect new Team constructor

cwendt94 · Oct 25, 2023 · 1910e15 · 1910e15
1 parent fa6bb2c
commit 1910e15
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/hockey/unit/test_team.py b/tests/hockey/unit/test_team.py
@@ -15,13 +15,14 @@ def setUp(self) -> None:
 
             self.team = self.data['teams'][3]
             self.team_roster = self.team['roster']
+            self.owners = self.team['owners']
 
     def test_team(self):
-        team = Team(self.team, roster= self.team_roster, schedule= self.schedule, year= self.year)
+        team = Team(self.team, roster= self.team_roster, schedule= self.schedule, year= self.year, owners=self.owners)
         self.assertEqual(team.team_abbrev, 'ESPC')
 
     def test_team_roster_df(self):
-        team = Team(self.team, roster= self.team_roster, schedule= self.schedule, year= self.year)
+        team = Team(self.team, roster= self.team_roster, schedule= self.schedule, year= self.year, owners=self.owners)
 
         self.assertEqual(len(team.roster), 25)
         self.assertEqual(team.roster[0].name, 'Thomas Chabot')