125 lines
3.8 KiB
Python
125 lines
3.8 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
测试修复后的memory_points处理
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
|
|
# 添加src目录到Python路径
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))
|
|
|
|
from person_info.person_info import Person
|
|
|
|
def test_memory_points_with_none():
|
|
"""测试包含None值的memory_points处理"""
|
|
print("测试包含None值的memory_points处理...")
|
|
|
|
# 创建一个测试Person实例
|
|
person = Person(person_id="test_user_123")
|
|
|
|
# 模拟包含None值的memory_points
|
|
person.memory_points = [
|
|
"喜好:喜欢咖啡:1.0",
|
|
None, # 模拟None值
|
|
"性格:开朗:1.0",
|
|
None, # 模拟另一个None值
|
|
"兴趣:编程:1.0"
|
|
]
|
|
|
|
print(f"原始memory_points: {person.memory_points}")
|
|
|
|
# 测试get_all_category方法
|
|
try:
|
|
categories = person.get_all_category()
|
|
print(f"获取到的分类: {categories}")
|
|
print("✓ get_all_category方法正常工作")
|
|
except Exception as e:
|
|
print(f"✗ get_all_category方法出错: {e}")
|
|
return False
|
|
|
|
# 测试get_memory_list_by_category方法
|
|
try:
|
|
memories = person.get_memory_list_by_category("喜好")
|
|
print(f"获取到的喜好记忆: {memories}")
|
|
print("✓ get_memory_list_by_category方法正常工作")
|
|
except Exception as e:
|
|
print(f"✗ get_memory_list_by_category方法出错: {e}")
|
|
return False
|
|
|
|
# 测试del_memory方法
|
|
try:
|
|
deleted_count = person.del_memory("喜好", "喜欢咖啡")
|
|
print(f"删除的记忆点数量: {deleted_count}")
|
|
print(f"删除后的memory_points: {person.memory_points}")
|
|
print("✓ del_memory方法正常工作")
|
|
except Exception as e:
|
|
print(f"✗ del_memory方法出错: {e}")
|
|
return False
|
|
|
|
return True
|
|
|
|
def test_memory_points_empty():
|
|
"""测试空的memory_points处理"""
|
|
print("\n测试空的memory_points处理...")
|
|
|
|
person = Person(person_id="test_user_456")
|
|
person.memory_points = []
|
|
|
|
try:
|
|
categories = person.get_all_category()
|
|
print(f"空列表的分类: {categories}")
|
|
print("✓ 空列表处理正常")
|
|
except Exception as e:
|
|
print(f"✗ 空列表处理出错: {e}")
|
|
return False
|
|
|
|
try:
|
|
memories = person.get_memory_list_by_category("测试分类")
|
|
print(f"空列表的记忆: {memories}")
|
|
print("✓ 空列表分类查询正常")
|
|
except Exception as e:
|
|
print(f"✗ 空列表分类查询出错: {e}")
|
|
return False
|
|
|
|
return True
|
|
|
|
def test_memory_points_all_none():
|
|
"""测试全部为None的memory_points处理"""
|
|
print("\n测试全部为None的memory_points处理...")
|
|
|
|
person = Person(person_id="test_user_789")
|
|
person.memory_points = [None, None, None]
|
|
|
|
try:
|
|
categories = person.get_all_category()
|
|
print(f"全None列表的分类: {categories}")
|
|
print("✓ 全None列表处理正常")
|
|
except Exception as e:
|
|
print(f"✗ 全None列表处理出错: {e}")
|
|
return False
|
|
|
|
try:
|
|
memories = person.get_memory_list_by_category("测试分类")
|
|
print(f"全None列表的记忆: {memories}")
|
|
print("✓ 全None列表分类查询正常")
|
|
except Exception as e:
|
|
print(f"✗ 全None列表分类查询出错: {e}")
|
|
return False
|
|
|
|
return True
|
|
|
|
if __name__ == "__main__":
|
|
print("开始测试修复后的memory_points处理...")
|
|
|
|
success = True
|
|
success &= test_memory_points_with_none()
|
|
success &= test_memory_points_empty()
|
|
success &= test_memory_points_all_none()
|
|
|
|
if success:
|
|
print("\n🎉 所有测试通过！memory_points的None值处理已修复。")
|
|
else:
|
|
print("\n❌ 部分测试失败，需要进一步检查。")
|