cognee-graph 0.1.3

Knowledge-graph database abstraction (embedded Ladybug) for the cognee pipeline.
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
55
56
57
58
59
60
61
62
63
64
65
66
67

#![allow(
    clippy::unwrap_used,
    clippy::expect_used,
    reason = "test code — panics are acceptable failures"
)]
//! Integration tests for `LadybugAdapter` using the shared GraphDBTrait test suite.
//!
//! These tests run against an embedded Ladybug database in a temporary directory
//! and require no external services.
#![cfg(feature = "ladybug")]

mod common;

use cognee_graph::{GraphDBTrait, LadybugAdapter};
use serial_test::serial;
use tempfile::TempDir;

/// Create a fresh Ladybug adapter backed by a temp directory.
async fn make_adapter() -> (LadybugAdapter, TempDir) {
    let dir = TempDir::new().expect("failed to create temp dir");
    let db_path = dir.path().join("test.db");
    let adapter = LadybugAdapter::new(db_path.to_str().unwrap())
        .await
        .expect("failed to create LadybugAdapter");
    adapter
        .initialize()
        .await
        .expect("failed to initialise Ladybug");
    (adapter, dir)
}

macro_rules! ladybug_test {
    ($name:ident) => {
        #[tokio::test]
        #[serial]
        async fn $name() {
            let (adapter, _dir) = make_adapter().await;
            common::$name(&adapter).await;
        }
    };
}

ladybug_test!(test_initialize_is_empty);
ladybug_test!(test_add_and_get_node);
ladybug_test!(test_add_nodes_batch);
ladybug_test!(test_has_node);
ladybug_test!(test_get_nodes_batch);
ladybug_test!(test_delete_node);
ladybug_test!(test_delete_nodes_batch);
ladybug_test!(test_node_upsert_same_id);
ladybug_test!(test_add_and_has_edge);
ladybug_test!(test_add_edges_batch);
ladybug_test!(test_edge_upsert_same_key);
ladybug_test!(test_has_edges);
ladybug_test!(test_has_edges_batch_equivalence);
ladybug_test!(test_get_edges);
ladybug_test!(test_get_neighbors);
ladybug_test!(test_get_connections);
ladybug_test!(test_get_graph_data);
ladybug_test!(test_get_graph_metrics);
ladybug_test!(test_get_filtered_graph_data);
ladybug_test!(test_get_nodeset_subgraph_or);
ladybug_test!(test_get_nodeset_subgraph_and);
ladybug_test!(test_get_id_filtered_graph_data);
ladybug_test!(test_delete_graph);
ladybug_test!(test_node_delete_cascades_edges);
ladybug_test!(test_properties_json_round_trip);