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

20
src/backend/tests/unit/test_process.py
View file

@ -282,12 +282,16 @@ async def test_load_langchain_object_with_cached_session(basic_graph_data):
# session_service = get_session_service()
# session_id1 = "non-existent-session-id"
# session_id = session_service.build_key(session_id1, basic_graph_data)
# graph1, artifacts1 = await session_service.load_session(session_id, data_graph=basic_graph_data, flow_id="flow_id")
# graph1, artifacts1 = await session_service.load_session(
# session_id, data_graph=basic_graph_data, flow_id="flow_id"
# )
# # Clear the cache
# await session_service.clear_session(session_id)
# # Use the new session_id to get the graph again
# graph2, artifacts2 = await session_service.load_session(session_id, data_graph=basic_graph_data, flow_id="flow_id")
# graph2, artifacts2 = await session_service.load_session(
# session_id, data_graph=basic_graph_data, flow_id="flow_id"
# )
#
# # Since the cache was cleared, objects should be different
# assert id(graph1) != id(graph2)
@ -297,8 +301,12 @@ async def test_load_langchain_object_with_cached_session(basic_graph_data):
# # Provide a non-existent session_id
# session_service = get_session_service()
# session_id1 = None
# graph1, artifacts1 = await session_service.load_session(session_id1, data_graph=basic_graph_data, flow_id="flow_id")
# graph1, artifacts1 = await session_service.load_session(
# session_id1, data_graph=basic_graph_data, flow_id="flow_id"
# )
# # Use the new session_id to get the langchain_object again
# graph2, artifacts2 = await session_service.load_session(session_id1, data_graph=basic_graph_data, flow_id="flow_id")
# graph2, artifacts2 = await session_service.load_session(
# session_id1, data_graph=basic_graph_data, flow_id="flow_id"
# )
#
# assert graph1 == graph2