Function test_embedding_crud

atomic-core/src/pristine/txn/write/embeddings.rs:253–268

()

    #[test]
    fn test_embedding_crud() {
        let dir = tempdir().unwrap();
        let pristine = Pristine::open(dir.path().join("db")).unwrap();
        let mut txn = pristine.write_txn().unwrap();
        txn.init_embeddings().unwrap();

        let record = make_record(384, 0);
        txn.put_embedding("memory/arch.md", 0, &record).unwrap();

        let retrieved = txn.get_embedding("memory/arch.md", 0).unwrap().unwrap();
        assert_eq!(retrieved.vector.len(), 384);
        assert_eq!(retrieved.preview, "chunk 0");

        assert!(txn.get_embedding("nonexistent", 0).unwrap().is_none());
        txn.commit().unwrap();
    }

    #[test]
    fn test_embedding_list_chunks() {