simple refactor

waketzheng · Jan 23, 2025 · 4e05120 · 4e05120
1 parent 766b109
commit 4e05120
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/conftest.py b/conftest.py
@@ -2,7 +2,7 @@
 
 import pytest
 
-from tortoise.contrib.test import finalizer, initializer
+from tortoise.contrib.test import MEMORY_SQLITE, finalizer, initializer
 
 
 @pytest.fixture(scope="session", autouse=True)
@@ -15,6 +15,6 @@ def initialize_tests(request):
     except ImportError:
         pass
 
-    db_url = os.getenv("TORTOISE_TEST_DB", "sqlite://:memory:")
+    db_url = os.getenv("TORTOISE_TEST_DB", MEMORY_SQLITE)
     initializer(["tests.testmodels"], db_url=db_url)
     request.addfinalizer(finalizer)