def test_table_creation(self): self.db.create_table('cards', OrderedDict( id = integer(primary_key=True), name = string(), def_ = integer(), atk = integer() )) column_names = [column.name for column in self.db.cards.columns] self.assertIn('cards', self.db.tables) self.assertEqual(sorted(column_names), ['atk', 'def', 'id', 'name'])
def test_creation_from_db(self): db = Database('sqlite:///:memory:') db.create_table('cards', OrderedDict( id = integer(primary_key=True), name = string(), atk = integer(), def_ = integer() )) Card = Item.from_table('Card', db.cards, exclude=['id']) self.assertEqual(Card.ATTRIBUTES, ('atk', 'def', 'name'))