diff --git a/bindings/python/tests/test_database.py b/bindings/python/tests/test_database.py index e52e21a26..9190cafea 100644 --- a/bindings/python/tests/test_database.py +++ b/bindings/python/tests/test_database.py @@ -1,9 +1,15 @@ import os import sqlite3 - import pytest import turso +def connect(provider, database): + if provider == "turso": + return turso.connect(database) + if provider == "sqlite3": + return sqlite3.connect(database) + raise Exception(f"Provider `{provider}` is not supported") + @pytest.fixture(autouse=True) def setup_database(): @@ -206,10 +212,3 @@ def test_with_statement(provider): assert max_id assert max_id == (2,) - -def connect(provider, database): - if provider == "turso": - return turso.connect(database) - if provider == "sqlite3": - return sqlite3.connect(database) - raise Exception(f"Provider `{provider}` is not supported")