diff --git a/tests/conftest.py b/tests/conftest.py index 7fb932ed8..0033b0652 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -13,6 +13,10 @@ import pytest from dotenv import load_dotenv from fastapi.testclient import TestClient from httpx import AsyncClient +from sqlmodel import Session, SQLModel, create_engine, select +from sqlmodel.pool import StaticPool +from typer.testing import CliRunner + from langflow.graph.graph.base import Graph from langflow.initial_setup.setup import STARTER_FOLDER_NAME from langflow.services.auth.utils import get_password_hash @@ -22,9 +26,6 @@ from langflow.services.database.models.folder.model import Folder from langflow.services.database.models.user.model import User, UserCreate from langflow.services.database.utils import session_getter from langflow.services.deps import get_db_service -from sqlmodel import Session, SQLModel, create_engine, select -from sqlmodel.pool import StaticPool -from typer.testing import CliRunner if TYPE_CHECKING: from langflow.services.database.service import DatabaseService @@ -37,7 +38,7 @@ def pytest_configure(config): config.addinivalue_line("markers", "noclient: don't create a client for this test") config.addinivalue_line("markers", "load_flows: load the flows for this test") config.addinivalue_line("markers", "api_key_required: run only if the api key is set in the environment variables") - data_path = Path(__file__).parent.parent.absolute() / "data" + data_path = Path(__file__).parent.absolute() / "data" pytest.BASIC_EXAMPLE_PATH = data_path / "basic_example.json" pytest.COMPLEX_EXAMPLE_PATH = data_path / "complex_example.json"