Skip to content

Commit ddae1e4

Browse files
style: apply ruff format to tests/test_drizzle.py
1 parent 1d74362 commit ddae1e4

1 file changed

Lines changed: 24 additions & 6 deletions

File tree

tests/test_drizzle.py

Lines changed: 24 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Tests for Drizzle ORM parser and generator."""
2+
23
from __future__ import annotations
34

45
import sys
@@ -135,9 +136,20 @@ def test_generate_simple_table(self):
135136

136137
schema = Schema()
137138
table = Table(name="users")
138-
table.columns.append(Column(name="id", type=ColumnType.INTEGER, primary_key=True))
139-
table.columns.append(Column(name="name", type=ColumnType.STRING, nullable=False, type_args={"length": 255}))
140-
table.columns.append(Column(name="email", type=ColumnType.TEXT, nullable=False, unique=True))
139+
table.columns.append(
140+
Column(name="id", type=ColumnType.INTEGER, primary_key=True)
141+
)
142+
table.columns.append(
143+
Column(
144+
name="name",
145+
type=ColumnType.STRING,
146+
nullable=False,
147+
type_args={"length": 255},
148+
)
149+
)
150+
table.columns.append(
151+
Column(name="email", type=ColumnType.TEXT, nullable=False, unique=True)
152+
)
141153
schema.tables.append(table)
142154

143155
gen = DrizzleGenerator(dialect="pg")
@@ -157,7 +169,9 @@ def test_generate_with_enum(self):
157169
schema = Schema()
158170
schema.enums.append(EnumType(name="role", values=["admin", "user"]))
159171
table = Table(name="users")
160-
table.columns.append(Column(name="id", type=ColumnType.INTEGER, primary_key=True))
172+
table.columns.append(
173+
Column(name="id", type=ColumnType.INTEGER, primary_key=True)
174+
)
161175
schema.tables.append(table)
162176

163177
gen = DrizzleGenerator(dialect="pg")
@@ -172,7 +186,9 @@ def test_generate_mysql_dialect(self):
172186

173187
schema = Schema()
174188
table = Table(name="products")
175-
table.columns.append(Column(name="id", type=ColumnType.INTEGER, primary_key=True))
189+
table.columns.append(
190+
Column(name="id", type=ColumnType.INTEGER, primary_key=True)
191+
)
176192
schema.tables.append(table)
177193

178194
gen = DrizzleGenerator(dialect="mysql")
@@ -186,7 +202,9 @@ def test_generate_default_now(self):
186202

187203
schema = Schema()
188204
table = Table(name="logs")
189-
table.columns.append(Column(name="createdAt", type=ColumnType.DATETIME, default="now()"))
205+
table.columns.append(
206+
Column(name="createdAt", type=ColumnType.DATETIME, default="now()")
207+
)
190208
schema.tables.append(table)
191209

192210
gen = DrizzleGenerator(dialect="pg")

0 commit comments

Comments
 (0)