class TestPMDataObject(TestCase):
def setUp(self):
db.init(':memory:')
db.create_tables([ PMDataObject ])
self.data_object = PMDataObject()
self.data_object.save()
def test_creation(self):
self.assertIsNotNone(self.data_object.id)
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 setUp(self):
db.init(':memory:')
db.create_tables([ PMDataObject ])
self.data_object = PMDataObject()
self.data_object.save()