1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
| package graph
|
| import (
| "testing"
|
| "github.com/aisim/kb-cli/internal/vault"
| )
|
| func TestBuildGraph(t *testing.T) {
| files := []*vault.FileMeta{
| {
| Path: "FAQ/充装类/001-test.md",
| Title: "测试FAQ",
| Section: "FAQ",
| Tags: []string{"充装", "智能枪"},
| Entities: []string{"YSP-35.5"},
| Wikilinks: []string{},
| Content: "测试内容",
| },
| {
| Path: "知识/002-配置.md",
| Title: "充装规格配置",
| Section: "知识",
| Tags: []string{"充装"},
| Entities: []string{},
| Wikilinks: []string{},
| Content: "配置内容",
| },
| }
|
| g := BuildGraph(files)
|
| if len(g.Nodes) != 2 {
| t.Errorf("node count = %d, want 2", len(g.Nodes))
| }
|
| // 应该有 3 条边：2条 tag + 1条 entity
| tagEdges := 0
| entityEdges := 0
| for _, e := range g.Edges {
| if e.Relation == "tag" {
| tagEdges++
| }
| if e.Relation == "entity" {
| entityEdges++
| }
| }
| if tagEdges != 3 { // 充装 + 智能枪 + 充装(第二个文件)
| t.Errorf("tag edges = %d, want 3", tagEdges)
| }
| if entityEdges != 1 {
| t.Errorf("entity edges = %d, want 1", entityEdges)
| }
| }
|
|
