mirror of
https://github.com/slhaf/Partner.git
synced 2026-05-12 16:53:04 +08:00
实现了MemoryGraph的序列化/反序列化;
实现了MemoryGraph的插入功能;
This commit is contained in:
132
src/test/java/memory/InsertTest.java
Normal file
132
src/test/java/memory/InsertTest.java
Normal file
@@ -0,0 +1,132 @@
|
||||
package memory;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import work.slhaf.memory.MemoryGraph;
|
||||
import work.slhaf.memory.content.MemorySlice;
|
||||
import work.slhaf.memory.node.MemoryNode;
|
||||
import work.slhaf.memory.node.TopicNode;
|
||||
|
||||
import java.time.LocalDateTime;
|
||||
import java.util.Arrays;
|
||||
import java.util.HashMap;
|
||||
import java.util.LinkedList;
|
||||
import java.util.List;
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
|
||||
public class InsertTest {
|
||||
private MemoryGraph memoryGraph;
|
||||
private final String testId = "test";
|
||||
|
||||
@Before
|
||||
public void setUp() {
|
||||
memoryGraph = new MemoryGraph(testId);
|
||||
memoryGraph.setTopicNodes(new HashMap<>());
|
||||
memoryGraph.setExistedTopics(new HashMap<>());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInsertMemory_NewRootTopic() {
|
||||
// 准备测试数据
|
||||
List<String> topicPath = new LinkedList<>(Arrays.asList("Programming", "Java", "Collections"));
|
||||
MemorySlice slice = createTestMemorySlice("slice1");
|
||||
|
||||
// 执行测试
|
||||
memoryGraph.insertMemory(topicPath, slice);
|
||||
|
||||
// 验证结果
|
||||
assertTrue(memoryGraph.getTopicNodes().containsKey("Programming"));
|
||||
TopicNode programmingNode = memoryGraph.getTopicNodes().get("Programming");
|
||||
|
||||
assertTrue(programmingNode.getTopicNodes().containsKey("Java"));
|
||||
TopicNode javaNode = programmingNode.getTopicNodes().get("Java");
|
||||
|
||||
assertTrue(javaNode.getTopicNodes().containsKey("Collections"));
|
||||
TopicNode collectionsNode = javaNode.getTopicNodes().get("Collections");
|
||||
|
||||
assertEquals(1, collectionsNode.getMemoryNodes().size());
|
||||
MemoryNode memoryNode = collectionsNode.getMemoryNodes().get(0);
|
||||
assertEquals(LocalDateTime.now().toLocalDate(), memoryNode.getLocalDateTime().toLocalDate());
|
||||
assertEquals(1, memoryNode.getMemorySliceList().size());
|
||||
assertEquals(slice, memoryNode.getMemorySliceList().get(0));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInsertMemory_ExistingTopicPath() {
|
||||
// 准备初始数据
|
||||
List<String> topicPath1 = new LinkedList<>(Arrays.asList("Programming", "Java", "Collections"));
|
||||
MemorySlice slice1 = createTestMemorySlice("slice1");
|
||||
memoryGraph.insertMemory(topicPath1, slice1);
|
||||
|
||||
// 插入第二个记忆片段到相同路径
|
||||
List<String> topicPath2 = new LinkedList<>(Arrays.asList("Programming", "Java", "Collections"));
|
||||
MemorySlice slice2 = createTestMemorySlice("slice2");
|
||||
memoryGraph.insertMemory(topicPath2, slice2);
|
||||
|
||||
// 验证结果
|
||||
TopicNode collectionsNode = memoryGraph.getTopicNodes().get("Programming")
|
||||
.getTopicNodes().get("Java")
|
||||
.getTopicNodes().get("Collections");
|
||||
|
||||
assertEquals(1, collectionsNode.getMemoryNodes().size()); // 同一天应该只有一个MemoryNode
|
||||
assertEquals(2, collectionsNode.getMemoryNodes().get(0).getMemorySliceList().size()); // 但有两个MemorySlice
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInsertMemory_DifferentDays() {
|
||||
// 准备测试数据
|
||||
List<String> topicPath = new LinkedList<>(Arrays.asList("Math", "Algebra"));
|
||||
MemorySlice slice1 = createTestMemorySlice("slice1");
|
||||
MemorySlice slice2 = createTestMemorySlice("slice2");
|
||||
|
||||
// 第一次插入
|
||||
memoryGraph.insertMemory(topicPath, slice1);
|
||||
|
||||
// 模拟第二天
|
||||
MemoryNode firstNode = memoryGraph.getTopicNodes().get("Math")
|
||||
.getTopicNodes().get("Algebra")
|
||||
.getMemoryNodes().get(0);
|
||||
firstNode.setLocalDateTime(LocalDateTime.now().minusDays(1));
|
||||
|
||||
// 第二次插入
|
||||
memoryGraph.insertMemory(topicPath, slice2);
|
||||
|
||||
// 验证结果
|
||||
TopicNode algebraNode = memoryGraph.getTopicNodes().get("Math")
|
||||
.getTopicNodes().get("Algebra");
|
||||
|
||||
assertEquals(2, algebraNode.getMemoryNodes().size()); // 应该有两个MemoryNode
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInsertMemory_PartialExistingPath() {
|
||||
// 准备初始数据 - 创建部分路径
|
||||
List<String> topicPath1 = new LinkedList<>(Arrays.asList("Science", "Physics"));
|
||||
MemorySlice slice1 = createTestMemorySlice("slice1");
|
||||
memoryGraph.insertMemory(topicPath1, slice1);
|
||||
|
||||
// 插入到已存在路径的扩展路径
|
||||
List<String> topicPath2 = new LinkedList<>(Arrays.asList("Science", "Physics", "Mechanics"));
|
||||
MemorySlice slice2 = createTestMemorySlice("slice2");
|
||||
memoryGraph.insertMemory(topicPath2, slice2);
|
||||
|
||||
// 验证结果
|
||||
TopicNode physicsNode = memoryGraph.getTopicNodes().get("Science")
|
||||
.getTopicNodes().get("Physics");
|
||||
|
||||
assertTrue(physicsNode.getTopicNodes().containsKey("Mechanics"));
|
||||
assertEquals(1, physicsNode.getMemoryNodes().size()); // Physics节点有自己的记忆
|
||||
assertEquals(1, physicsNode.getTopicNodes().get("Mechanics").getMemoryNodes().size()); // Mechanics节点也有记忆
|
||||
}
|
||||
|
||||
private MemorySlice createTestMemorySlice(String id) {
|
||||
MemorySlice slice = new MemorySlice();
|
||||
slice.setMemoryId(id);
|
||||
slice.setMemoryRank(1);
|
||||
// 可以设置其他必要属性
|
||||
return slice;
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user