mirror of https://github.com/Mai-with-u/MaiBot.git
final
SengokuCola
parent
794a0d8fd4
commit
c28e964747
|
|
@ -1,73 +0,0 @@
|
|||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
测试del_memory函数的脚本
|
||||
"""
|
||||
|
||||
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_del_memory():
|
||||
"""测试del_memory函数"""
|
||||
print("开始测试del_memory函数...")
|
||||
|
||||
# 创建一个测试用的Person实例（不连接数据库）
|
||||
person = Person.__new__(Person)
|
||||
person.person_id = "test_person"
|
||||
person.memory_points = [
|
||||
"性格:这个人很友善:5.0",
|
||||
"性格:这个人很友善:4.0",
|
||||
"爱好:喜欢打游戏:3.0",
|
||||
"爱好:喜欢打游戏:2.0",
|
||||
"工作:是一名程序员:1.0",
|
||||
"性格:这个人很友善:6.0"
|
||||
]
|
||||
|
||||
print(f"原始记忆点数量: {len(person.memory_points)}")
|
||||
print("原始记忆点:")
|
||||
for i, memory in enumerate(person.memory_points):
|
||||
print(f" {i+1}. {memory}")
|
||||
|
||||
# 测试删除"性格"分类中"这个人很友善"的记忆
|
||||
print("\n测试1: 删除'性格'分类中'这个人很友善'的记忆")
|
||||
deleted_count = person.del_memory("性格", "这个人很友善")
|
||||
print(f"删除了 {deleted_count} 个记忆点")
|
||||
print("删除后的记忆点:")
|
||||
for i, memory in enumerate(person.memory_points):
|
||||
print(f" {i+1}. {memory}")
|
||||
|
||||
# 测试删除"爱好"分类中"喜欢打游戏"的记忆
|
||||
print("\n测试2: 删除'爱好'分类中'喜欢打游戏'的记忆")
|
||||
deleted_count = person.del_memory("爱好", "喜欢打游戏")
|
||||
print(f"删除了 {deleted_count} 个记忆点")
|
||||
print("删除后的记忆点:")
|
||||
for i, memory in enumerate(person.memory_points):
|
||||
print(f" {i+1}. {memory}")
|
||||
|
||||
# 测试相似度匹配
|
||||
print("\n测试3: 测试相似度匹配")
|
||||
person.memory_points = [
|
||||
"性格:这个人非常友善:5.0",
|
||||
"性格:这个人很友善:4.0",
|
||||
"性格:这个人友善:3.0"
|
||||
]
|
||||
print("原始记忆点:")
|
||||
for i, memory in enumerate(person.memory_points):
|
||||
print(f" {i+1}. {memory}")
|
||||
|
||||
# 删除"这个人很友善"（应该匹配"这个人很友善"和"这个人友善"）
|
||||
deleted_count = person.del_memory("性格", "这个人很友善", similarity_threshold=0.8)
|
||||
print(f"删除了 {deleted_count} 个记忆点")
|
||||
print("删除后的记忆点:")
|
||||
for i, memory in enumerate(person.memory_points):
|
||||
print(f" {i+1}. {memory}")
|
||||
|
||||
print("\n测试完成!")
|
||||
|
||||
if __name__ == "__main__":
|
||||
test_del_memory()
|
||||
|
|
@ -1,124 +0,0 @@
|
|||
#!/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❌ 部分测试失败，需要进一步检查。")
|
||||
Loading…
Reference in New Issue