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

ref: Add ALL ruff rules for tests (#4183)

Browse source 
Add ALL ruff rules for tests
This commit is contained in:
Christophe Bornet 2024-10-19 22:41:37 +02:00 • committed by GitHub
commit f96f2eaf8a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
67 changed files with 421 additions and 361 deletions
Download patch file Download diff file Expand all files Collapse all files

25
src/backend/tests/integration/utils.py
View file

@ -12,26 +12,26 @@ from langflow.graph import Graph
from langflow.processing.process import run_graph_internal
def check_env_vars(*vars):
def check_env_vars(*env_vars):
"""Check if all specified environment variables are set.
Args:
*vars (str): The environment variables to check.
*env_vars (str): The environment variables to check.
Returns:
bool: True if all environment variables are set, False otherwise.
"""
return all(os.getenv(var) for var in vars)
return all(os.getenv(var) for var in env_vars)
def valid_nvidia_vectorize_region(api_endpoint: str) -> bool:
"""Check if the specified region is valid.
Args:
region (str): The region to check.
api_endpoint: The API endpoint to check.
Returns:
bool: True if the region is contains hosted nvidia models, False otherwise.
True if the region contains hosted nvidia models, False otherwise.
"""
parsed_endpoint = parse_api_endpoint(api_endpoint)
if not parsed_endpoint:
@ -63,12 +63,12 @@ class JSONFlow:
json: dict
def get_components_by_type(self, component_type):
result = []
for node in self.json["data"]["nodes"]:
if node["data"]["type"] == component_type:
result.append(node["id"])
result = [node["id"] for node in self.json["data"]["nodes"] if node["data"]["type"] == component_type]
if not result:
msg = f"Component of type {component_type} not found, available types: {', '.join({node['data']['type'] for node in self.json['data']['nodes']})}"
msg = (
f"Component of type {component_type} not found, "
f"available types: {', '.join({node['data']['type'] for node in self.json['data']['nodes']})}"
)
raise ValueError(msg)
return result
@ -97,7 +97,8 @@ class JSONFlow:
def download_flow_from_github(name: str, version: str) -> JSONFlow:
response = requests.get(
f"https://raw.githubusercontent.com/langflow-ai/langflow/v{version}/src/backend/base/langflow/initial_setup/starter_projects/{name}.json"
f"https://raw.githubusercontent.com/langflow-ai/langflow/v{version}/src/backend/base/langflow/initial_setup/starter_projects/{name}.json",
timeout=10,
)
response.raise_for_status()
as_json = response.json()
@ -151,7 +152,7 @@ async def run_single_component(
raw_inputs[key] = value
if isinstance(value, Component):
msg = "Component inputs must be wrapped in ComponentInputHandle"
raise ValueError(msg)
raise TypeError(msg)
component = clazz(**raw_inputs, _user_id=user_id)
component_id = graph.add_component(component)
if inputs: