test_query_neo4j_graph_v2.py 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364
  1. #!/usr/bin/env python3
  2. # -*- coding: utf-8 -*-
  3. """
  4. 测试优化后的 query_neo4j_graph 函数
  5. """
  6. import sys
  7. import os
  8. import logging
  9. # 添加项目根目录到Python路径
  10. sys.path.append(os.path.dirname(os.path.abspath(__file__)))
  11. # 配置日志
  12. logging.basicConfig(
  13. level=logging.INFO,
  14. format='%(asctime)s - %(levelname)s - %(name)s - %(funcName)s - %(lineno)d - %(message)s'
  15. )
  16. def test_query_neo4j_graph():
  17. """测试优化后的 query_neo4j_graph 函数"""
  18. try:
  19. from app.core.data_parse.parse_system import query_neo4j_graph
  20. # 测试查询文本
  21. query_text = "大连硬石酒店"
  22. print(f"测试查询文本: {query_text}")
  23. print("-" * 50)
  24. # 调用函数
  25. result = query_neo4j_graph(query_text)
  26. # 打印结果
  27. print("函数执行结果:")
  28. print(f"状态码: {result.get('code')}")
  29. print(f"成功: {result.get('success')}")
  30. print(f"消息: {result.get('message')}")
  31. print(f"匹配的标签: {result.get('matched_labels')}")
  32. print(f"匹配的酒店: {result.get('matched_hotels')}")
  33. print(f"非空字段: {result.get('non_empty_fields')}")
  34. print(f"查询语句: {result.get('query')}")
  35. print(f"数据条数: {len(result.get('data', []))}")
  36. if result.get('data'):
  37. print("\n前5条数据:")
  38. for i, record in enumerate(result.get('data', [])[:5]):
  39. print(f" {i+1}. {record}")
  40. return result
  41. except Exception as e:
  42. print(f"测试失败: {str(e)}")
  43. import traceback
  44. traceback.print_exc()
  45. return None
  46. if __name__ == "__main__":
  47. print("开始测试优化后的 query_neo4j_graph 函数...")
  48. result = test_query_neo4j_graph()
  49. if result and result.get('success'):
  50. print("\n✅ 测试成功!函数工作正常。")
  51. else:
  52. print("\n❌ 测试失败!函数存在问题。")