jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🔧 refactor(conftest.py): remove unnecessary comments for cleaner code

Browse source 
🔧 refactor(conftest.py): reformat code for better readability
🔧 refactor(conftest.py): move import statements to new lines for better readability
This commit is contained in:
gustavoschaedler 2023-07-14 19:19:42 +01:00
commit 7c37c6d74f
1 changed files with 15 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/conftest.py
View file

@ -93,8 +93,8 @@ def json_flow():
return f.read() return f.read()
@pytest.fixture(name="session") # @pytest.fixture(name="session")
def session_fixture(): # def session_fixture():
engine = create_engine( engine = create_engine(
"sqlite://", connect_args={"check_same_thread": False}, poolclass=StaticPool "sqlite://", connect_args={"check_same_thread": False}, poolclass=StaticPool
) )
@ -103,19 +103,19 @@ def session_fixture(): #
yield session yield session
@pytest.fixture(name="client") # @pytest.fixture(name="client")
def client_fixture(session: Session): # def client_fixture(session: Session):
def get_session_override(): # def get_session_override():
return session return session
from langflow.main import create_app from langflow.main import create_app
app = create_app() app = create_app()
app.dependency_overrides[get_session] = get_session_override # app.dependency_overrides[get_session] = get_session_override
yield TestClient(app) yield TestClient(app)
app.dependency_overrides.clear() # app.dependency_overrides.clear()
@pytest.fixture @pytest.fixture
@ -311,7 +311,8 @@ class DocumentFilterByLengthComponent(CustomComponent):
@pytest.fixture @pytest.fixture
def get_request(): def get_request():
return """import requests return """
import requests
from typing import Dict, Union from typing import Dict, Union
from langchain.schema import Document from langchain.schema import Document
from langflow.interface.custom.base import CustomComponent 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 # Create a document with the response text and the URL as metadata
document = Document(page_content=response.text, metadata={"url": url}) document = Document(page_content=response.text, metadata={"url": url})
return document""" return document
"""
@pytest.fixture @pytest.fixture
def post_request(): def post_request():
return """import requests return """
import requests
from typing import Dict, Union from typing import Dict, Union
from langchain.schema import Document from langchain.schema import Document
from langflow.interface.custom.base import CustomComponent 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 # 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}) document = Document(page_content=response.text, metadata={"url": url, "data": data})
return document""" return document
"""