def setUp(self):
# TODO Lots of redundancy here with main app and other tests. Can we
# abstract?
db.init(':memory:')
classes = PMDataObject.__subclasses__()
db.create_tables(classes)
for class_ in classes:
if class_.__name__ not in PMDataObject.deferred_relations:
PMDataObject.deferred_relations[class_.__name__] = class_
league = League.create(name="Band Battle")
season = Season.create(league=league,
start_year=2016,
end_year=2017,
is_current=True)
self.team1 = Team.create(name="Sex Bob-omb",
city="Toronto",
skill=90,
abbreviation="SBO",
league=league)
TeamStats.create(team=self.team1, season=season)
self.team2 = Team.create(name="The Clash at Demonhead",
city="New York",
skill=90,
abbreviation="TCD",
league=league)
TeamStats.create(team=self.team2, season=season)
def current_team_season_stats(self):
"""Return the PlayerStats object for the current season and team"""
stats = PMDataObject.deferred_relations['PlayerStats']
current_season = Season.select(Season.id).where(Season.is_current==True).get()
instance, create = stats.get_or_create(team=self.team,
season=current_season, player=self)
return instance
def setUp(self):
db.init(':memory:')
classes = [Team, League, Season, Person, Player, PlayerStats]
db.create_tables(classes)
for class_ in classes:
if class_.__name__ not in PMDataObject.deferred_relations:
PMDataObject.deferred_relations[class_.__name__] = class_
self.league = League.create(name="Band Battle")
self.season = Season.create(league=self.league,
start_year=2016,
end_year=2017,
is_current=True)
self.team = Team.create(name="Sex Bob-omb",
city="Toronto",
skill=90,
abbreviation="TOR",
league=self.league)
self.person = Person.create(forename="Scott", surname="Pilgrim")
self.player = Player.create(person=self.person,
team=self.team,
position="F",
scoring_rate=12.0,
shot_rate=12.0,
)
self.stats = PlayerStats.create(player=self.player,
season=self.season,
team=self.team)
def new_season(league):
"""Start a new season"""
old_season = league.current_season
if old_season is None:
new_season = Season.create(league=league,
start_year=2016,
end_year=2017,
is_current=True)
else:
old_season.is_current = False
old_season.save()
new_season = Season.create(league=league,
start_year=old_season.start_year + 1,
end_year=old_season.end_year + 1,
is_current=True)
for team in league.teams:
TeamStats.create(team=team, season=new_season)
def setUp(self):
db.init(':memory:')
db.create_tables([Team, League, Season])
PMDataObject.deferred_relations['Season'] = Season
self.league = League.create(name="Band Battle")
self.season = Season.create(league=self.league,
start_year=2016,
end_year=2017,
is_current=True)
self.team = Team.create(name="Sex Bob-omb",
city="Toronto",
skill=90,
abbreviation="TOR",
league=self.league)
def setUp(self):
db.init(':memory:')
classes = [Team, League, Season, TeamStats]
db.create_tables(classes)
for class_ in classes:
if class_.__name__ not in PMDataObject.deferred_relations:
PMDataObject.deferred_relations[class_.__name__] = class_
self.league = League.create(name="Band Battle")
self.season = Season.create(league=self.league,
start_year=2016,
end_year=2017,
is_current=True)
self.team = Team.create(name="Sex Bob-omb",
city="Toronto",
skill=90,
abbreviation="TOR",
league=self.league)
self.team_stats = TeamStats.create(team=self.team,
season=self.season)
def setUp(self):
db.init(':memory:')
classes = [Team, League, Season, PlayerStats, Person, Player]
db.create_tables(classes)
for class_ in classes:
if class_.__name__ not in PMDataObject.deferred_relations:
PMDataObject.deferred_relations[class_.__name__] = class_
self.league = League.create(name="NHL")
self.team = Team.create(name="Canucks",
city="Vancouver",
skill=100,
abbreviation="VAN",
league=self.league)
self.person = Person.create_from_name(name=Name(forename = "Cliff",
surname = "Ronning"))
self.player = Player.create(person=self.person, team = self.team,
position = "C", scoring_rate=0.07, shot_rate=6)
self.season = Season.create(league=self.league,
start_year=2016,
end_year=2017,
is_current=True)
def setUp(self):
db.init(':memory:')
db.create_tables([League, Season])
self.league = League.create(name="Band Battle")
self.season = Season.create(league=self.league, start_year=2016, end_year=2017)
