diff --git a/sqlite_utils/db.py b/sqlite_utils/db.py index 3a41434..7576141 100644 --- a/sqlite_utils/db.py +++ b/sqlite_utils/db.py @@ -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 diff --git a/tests/test_create.py b/tests/test_create.py index 6528a23..ddc45e5 100644 --- a/tests/test_create.py +++ b/tests/test_create.py @@ -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})