ext/python: Makes linter happy

This commit is contained in:
Diego Reis
2025-03-24 12:40:59 -03:00
parent 9a8970b6a8
commit 6edf3dd3b1
2 changed files with 14 additions and 9 deletions

View File

@@ -18,13 +18,16 @@ with limbo.connect("sqlite.db") as con:
("alice", "alice@example.com", "admin"), ("alice", "alice@example.com", "admin"),
("bob", "bob@example.com", "user"), ("bob", "bob@example.com", "user"),
("charlie", "charlie@example.com", "moderator"), ("charlie", "charlie@example.com", "moderator"),
("diana", "diana@example.com", "user") ("diana", "diana@example.com", "user"),
] ]
for username, email, role in sample_users: for username, email, role in sample_users:
cur.execute(""" cur.execute(
"""
INSERT INTO users (username, email, role) INSERT INTO users (username, email, role)
VALUES (?, ?, ?) VALUES (?, ?, ?)
""", (username, email, role)) """,
(username, email, role),
)
# Use commit to ensure the data is saved # Use commit to ensure the data is saved
con.commit() con.commit()

View File

@@ -144,6 +144,7 @@ def test_commit(provider):
conn.close() conn.close()
assert record assert record
@pytest.mark.parametrize("provider", ["sqlite3", "limbo"]) @pytest.mark.parametrize("provider", ["sqlite3", "limbo"])
def test_with_statement(provider): def test_with_statement(provider):
with connect(provider, "tests/database.db") as conn: with connect(provider, "tests/database.db") as conn:
@@ -156,6 +157,7 @@ def test_with_statement(provider):
assert max_id assert max_id
assert max_id == (2,) assert max_id == (2,)
def connect(provider, database): def connect(provider, database):
if provider == "limbo": if provider == "limbo":
return limbo.connect(database) return limbo.connect(database)