tinycortex 0.1.1

Rust core for the TinyCortex memory system
Documentation 
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

//! Tests for the SQLite-backed graph occurrence adapter.

use crate::memory::graph::EntityOccurrenceIndex;
use crate::memory::retrieval::graph_adapter::ConfigEntityIndex;
use crate::memory::retrieval::test_support::{fixed_ts, index_entity_occurrence, test_config};
use crate::memory::score::extract::EntityKind;

#[test]
fn config_entity_index_reads_nodes_and_entities_from_sqlite_index() {
    let (_tmp, cfg) = test_config();
    let ts = fixed_ts().timestamp_millis();
    index_entity_occurrence(
        &cfg,
        "topic:phoenix",
        EntityKind::Topic,
        "Phoenix",
        "node-a",
        "summary",
        ts,
        Some("tree:eng"),
    );
    index_entity_occurrence(
        &cfg,
        "person:alice",
        EntityKind::Person,
        "Alice",
        "node-a",
        "summary",
        ts,
        Some("tree:eng"),
    );
    index_entity_occurrence(
        &cfg,
        "topic:phoenix",
        EntityKind::Topic,
        "Phoenix",
        "node-b",
        "leaf",
        ts + 1,
        None,
    );

    let idx = ConfigEntityIndex::new(&cfg);
    let nodes = idx.nodes_for_entity("topic:phoenix").unwrap();
    assert_eq!(nodes, vec!["node-b".to_string(), "node-a".to_string()]);

    let entities = idx.entities_on_node("node-a").unwrap();
    assert_eq!(
        entities,
        vec!["person:alice".to_string(), "topic:phoenix".to_string()]
    );
    assert!(idx.nodes_for_entity("topic:missing").unwrap().is_empty());
    assert!(idx.entities_on_node("node-missing").unwrap().is_empty());
}