Modify the query tests to match the test DB

database/__init__.py

@@ -1,14 +1,13 @@
-from constants import DB
+from databases import Database
 from sqlalchemy import MetaData, create_engine
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker
-from databases import Database
+
+from constants import DB
 
 metadata = MetaData()
-engine = create_engine(DB)
+engine = create_engine(DB, connect_args={"check_same_thread": False})
 metadata.create_all(engine)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 Base = declarative_base()
 database = Database(DB)
-
-from database import models

database/crud.py

@@ -1,9 +1,9 @@
 from datetime import datetime
+from pydoc import locate
 
 from app import app
 from app.schemas import *
 from database import database
-from pydoc import locate
 
 
 @app.on_event("startup")