Format Python bindings code using Ruff

- Use double quotes (Ruff/Black default)
- Configure some set of linters to use with Ruff
This commit is contained in:
Lauri Virtanen 2024-08-31 13:43:49 +03:00
parent 4738d16c6f
commit 6bd1d28e26
4 changed files with 54 additions and 55 deletions

View file

@ -5,54 +5,48 @@ import pytest
import limbo
@pytest.mark.parametrize('provider', ['sqlite3', 'limbo'])
@pytest.mark.parametrize("provider", ["sqlite3", "limbo"])
def test_fetchall_select_all_users(provider):
conn = connect(provider, 'tests/database.db')
conn = connect(provider, "tests/database.db")
cursor = conn.cursor()
cursor.execute('SELECT * FROM users')
cursor.execute("SELECT * FROM users")
users = cursor.fetchall()
assert users
assert users == [(1, 'alice'), (2, 'bob')]
assert users == [(1, "alice"), (2, "bob")]
@pytest.mark.parametrize(
'provider',
[
'sqlite3',
'limbo'
],
)
@pytest.mark.parametrize("provider", ["sqlite3", "limbo"])
def test_fetchall_select_user_ids(provider):
conn = connect(provider, 'tests/database.db')
conn = connect(provider, "tests/database.db")
cursor = conn.cursor()
cursor.execute('SELECT id FROM users')
cursor.execute("SELECT id FROM users")
user_ids = cursor.fetchall()
assert user_ids
assert user_ids == [(1,), (2,)]
@pytest.mark.parametrize('provider', ['sqlite3', 'limbo'])
@pytest.mark.parametrize("provider", ["sqlite3", "limbo"])
def test_fetchone_select_all_users(provider):
conn = connect(provider, 'tests/database.db')
conn = connect(provider, "tests/database.db")
cursor = conn.cursor()
cursor.execute('SELECT * FROM users')
cursor.execute("SELECT * FROM users")
alice = cursor.fetchone()
assert alice
assert alice == (1, 'alice')
assert alice == (1, "alice")
bob = cursor.fetchone()
assert bob
assert bob == (2, 'bob')
assert bob == (2, "bob")
@pytest.mark.parametrize('provider', ['sqlite3', 'limbo'])
@pytest.mark.parametrize("provider", ["sqlite3", "limbo"])
def test_fetchone_select_max_user_id(provider):
conn = connect(provider, 'tests/database.db')
conn = connect(provider, "tests/database.db")
cursor = conn.cursor()
cursor.execute('SELECT MAX(id) FROM users')
cursor.execute("SELECT MAX(id) FROM users")
max_id = cursor.fetchone()
assert max_id
@ -60,8 +54,8 @@ def test_fetchone_select_max_user_id(provider):
def connect(provider, database):
if provider == 'limbo':
if provider == "limbo":
return limbo.connect(database)
if provider == 'sqlite3':
if provider == "sqlite3":
return sqlite3.connect(database)
raise Exception(f'Provider `{provider}` is not supported')
raise Exception(f"Provider `{provider}` is not supported")