kkachi 0.1.8

High-performance, zero-copy library for optimizing language model prompts and programs
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

// Property-based tests for Memory::upsert and upsert_tagged.

use kkachi::recursive::memory;
use proptest::prelude::*;

proptest! {
    #[test]
    fn upsert_idempotent_for_any_content(content in "\\PC{1,200}") {
        let mut mem = memory();
        let id1 = mem.upsert(&content).unwrap();
        let id2 = mem.upsert(&content).unwrap();
        prop_assert_eq!(id1, id2, "same content must always produce the same ID");
    }

    #[test]
    fn upsert_tagged_preserves_tag(
        tag in "[a-z]{1,20}",
        content in "\\PC{1,200}",
    ) {
        let mut mem = memory();
        mem.upsert_tagged(&tag, &content).unwrap();
        let tags = mem.tags().unwrap();
        prop_assert!(
            tags.contains(&tag),
            "tag {:?} must appear in tags() after upsert_tagged, got {:?}",
            tag,
            tags,
        );
    }
}