From c5286d2c66b8c187002f9ad8991a6665b2b092f9 Mon Sep 17 00:00:00 2001 From: Simon Willison Date: Sun, 24 Feb 2019 11:40:58 -0800 Subject: [PATCH] Black formatting --- sqlite_utils/db.py | 4 +--- tests/test_create.py | 18 +++++------------- 2 files changed, 6 insertions(+), 16 deletions(-) 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})