From 7c37c6d74fc1899a7e3c1bd79c99cce42fa37919 Mon Sep 17 00:00:00 2001 From: gustavoschaedler Date: Fri, 14 Jul 2023 19:19:42 +0100 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20refactor(conftest.py):=20remove?= =?UTF-8?q?=20unnecessary=20comments=20for=20cleaner=20code=20=F0=9F=94=A7?= =?UTF-8?q?=20refactor(conftest.py):=20reformat=20code=20for=20better=20re?= =?UTF-8?q?adability=20=F0=9F=94=A7=20refactor(conftest.py):=20move=20impo?= =?UTF-8?q?rt=20statements=20to=20new=20lines=20for=20better=20readability?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/conftest.py | 26 +++++++++++++++----------- 1 file changed, 15 insertions(+), 11 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index ae78483e6..8be738632 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -93,8 +93,8 @@ def json_flow(): return f.read() -@pytest.fixture(name="session") # -def session_fixture(): # +@pytest.fixture(name="session") +def session_fixture(): engine = create_engine( "sqlite://", connect_args={"check_same_thread": False}, poolclass=StaticPool ) @@ -103,19 +103,19 @@ def session_fixture(): # yield session -@pytest.fixture(name="client") # -def client_fixture(session: Session): # - def get_session_override(): # +@pytest.fixture(name="client") +def client_fixture(session: Session): + def get_session_override(): return session from langflow.main import create_app app = create_app() - app.dependency_overrides[get_session] = get_session_override # + app.dependency_overrides[get_session] = get_session_override yield TestClient(app) - app.dependency_overrides.clear() # + app.dependency_overrides.clear() @pytest.fixture @@ -311,7 +311,8 @@ class DocumentFilterByLengthComponent(CustomComponent): @pytest.fixture def get_request(): - return """import requests + return """ +import requests from typing import Dict, Union from langchain.schema import Document from langflow.interface.custom.base import CustomComponent @@ -333,12 +334,14 @@ class GetRequestComponent(CustomComponent): # Create a document with the response text and the URL as metadata document = Document(page_content=response.text, metadata={"url": url}) - return document""" + return document +""" @pytest.fixture def post_request(): - return """import requests + return """ +import requests from typing import Dict, Union from langchain.schema import Document from langflow.interface.custom.base import CustomComponent @@ -361,4 +364,5 @@ class PostRequestComponent(CustomComponent): # Create a document with the response text and the URL and data as metadata document = Document(page_content=response.text, metadata={"url": url, "data": data}) - return document""" + return document +"""