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'))
