remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-08-01 15:04:11 +02:00
Code Issues Projects Releases Packages Wiki Activity

Black formatting

Browse source
This commit is contained in:
Simon Willison 2019-02-24 11:40:58 -08:00
commit c5286d2c66
2 changed files with 6 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

4
sqlite_utils/db.py
View file

@ -257,9 +257,7 @@ class Table:
def add_column(self, col_name, col_type):
sql = "ALTER TABLE [{table}] ADD COLUMN [{col_name}] {col_type};".format(
table=self.name,
col_name=col_name,
col_type=COLUMN_TYPE_MAPPING[col_type],
table=self.name, col_name=col_name, col_type=COLUMN_TYPE_MAPPING[col_type]
)
self.db.conn.execute(sql)
return self

18
tests/test_create.py
View file

@ -126,19 +126,11 @@ def test_create_table_works_for_m2m_with_only_foreign_keys(fresh_db):
@pytest.mark.parametrize(
"col_name,col_type,expected_schema",
(
(
"nickname", str, "CREATE TABLE [dogs] ( [name] TEXT , [nickname] TEXT)"
),
(
"dob", datetime.date, "CREATE TABLE [dogs] ( [name] TEXT , [dob] TEXT)"
),
(
"age", int, "CREATE TABLE [dogs] ( [name] TEXT , [age] INTEGER)"
),
(
"weight", float, "CREATE TABLE [dogs] ( [name] TEXT , [weight] FLOAT)"
),
)
("nickname", str, "CREATE TABLE [dogs] ( [name] TEXT , [nickname] TEXT)"),
("dob", datetime.date, "CREATE TABLE [dogs] ( [name] TEXT , [dob] TEXT)"),
("age", int, "CREATE TABLE [dogs] ( [name] TEXT , [age] INTEGER)"),
("weight", float, "CREATE TABLE [dogs] ( [name] TEXT , [weight] FLOAT)"),
),
)
def test_add_column(fresh_db, col_name, col_type, expected_schema):
fresh_db.create_table("dogs", {"name": str})