gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix linting

Browse Source
This commit is contained in:
yangdx
2025-04-16 14:55:54 +08:00
parent d370c0ae12
commit 54f720cb27
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_graph_storage.py
View File

@@ -746,23 +746,30 @@ async def test_graph_batch_operations(storage):
# 检查是否包含与机器学习、自然语言处理和计算机视觉的连接（忽略方向）
has_connection_with_node2 = any(
(src == node2_id and tgt == node3_id) or (src == node3_id and tgt == node2_id)
(src == node2_id and tgt == node3_id)
or (src == node3_id and tgt == node2_id)
for src, tgt in nodes_edges[node3_id]
)
has_connection_with_node4 = any(
(src == node3_id and tgt == node4_id) or (src == node4_id and tgt == node3_id)
(src == node3_id and tgt == node4_id)
or (src == node4_id and tgt == node3_id)
for src, tgt in nodes_edges[node3_id]
)
has_connection_with_node5 = any(
(src == node3_id and tgt == node5_id) or (src == node5_id and tgt == node3_id)
(src == node3_id and tgt == node5_id)
or (src == node5_id and tgt == node3_id)
for src, tgt in nodes_edges[node3_id]
)
assert (
has_connection_with_node2
), f"节点 {node3_id} 的边列表中应包含与 {node2_id} 的连接"
assert has_connection_with_node4, f"节点 {node3_id} 的边列表中应包含与 {node4_id} 的连接"
assert has_connection_with_node5, f"节点 {node3_id} 的边列表中应包含与 {node5_id} 的连接"
assert (
has_connection_with_node4
), f"节点 {node3_id} 的边列表中应包含与 {node4_id} 的连接"
assert (
has_connection_with_node5
), f"节点 {node3_id} 的边列表中应包含与 {node5_id} 的连接"
print("无向图特性验证成功：批量获取的节点边包含所有相关的边（无论方向）")