nidx-binding 6.9.7.post702__tar.gz → 6.9.7.post704__tar.gz

{nidx_binding-6.9.7.post702 → nidx_binding-6.9.7.post704}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nidx_binding
-Version: 6.9.7.post702
+Version: 6.9.7.post704
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

{nidx_binding-6.9.7.post702 → nidx_binding-6.9.7.post704}/nidx_protos/pyproject.toml

@@ -10,7 +10,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "nidx_protos"
-version = "6.9.7.post702"
+version = "6.9.7.post704"
 license = "AGPL-3.0-or-later"
 description = "Protobuf definitions for nucliadb/nidx"
 authors = [{ name = "Nuclia", email = "nucliadb@nuclia.com" }]

{nidx_binding-6.9.7.post702 → nidx_binding-6.9.7.post704}/nidx_relation/tests/test_writer.rs

@@ -27,6 +27,8 @@ use nidx_protos::{
 use nidx_relation::{RelationConfig, RelationIndexer};
 use tempfile::TempDir;
 
+use crate::common::TestOpener;
+
 #[test]
 fn test_index_docs() -> anyhow::Result<()> {
     let dir = TempDir::new().unwrap();
@@ -82,3 +84,91 @@ fn test_index_docs() -> anyhow::Result<()> {
 
     Ok(())
 }
+
+#[test]
+fn test_merge() -> anyhow::Result<()> {
+    let dir1 = TempDir::new().unwrap();
+
+    let resource = Resource {
+        resource: Some(ResourceId {
+            uuid: "01808bbd8e784552967a4fb0d8b6e584".to_string(),
+            shard_id: "shard_id".to_string(),
+        }),
+        field_relations: HashMap::from([(
+            "a/metadata".to_string(),
+            IndexRelations {
+                relations: vec![common::create_relation(
+                    "01808bbd8e784552967a4fb0d8b6e584".to_string(),
+                    NodeType::Resource,
+                    "".to_string(),
+                    "dog".to_string(),
+                    NodeType::Entity,
+                    "ANIMALS".to_string(),
+                    RelationType::Entity,
+                    "IS".to_string(),
+                )],
+            },
+        )]),
+        ..Default::default()
+    };
+
+    let meta1 = RelationIndexer
+        .index_resource(dir1.path(), &RelationConfig::default(), &resource)?
+        .unwrap();
+
+    assert_eq!(meta1.records, 1);
+
+    let dir2 = TempDir::new().unwrap();
+
+    let resource = Resource {
+        resource: Some(ResourceId {
+            uuid: "81808bbd8e784552967a4fb0d8b6e584".to_string(),
+            shard_id: "shard_id".to_string(),
+        }),
+        field_relations: HashMap::from([(
+            "a/metadata".to_string(),
+            IndexRelations {
+                relations: vec![common::create_relation(
+                    "81808bbd8e784552967a4fb0d8b6e584".to_string(),
+                    NodeType::Resource,
+                    "".to_string(),
+                    "cat".to_string(),
+                    NodeType::Entity,
+                    "ANIMALS".to_string(),
+                    RelationType::Entity,
+                    "IS".to_string(),
+                )],
+            },
+        )]),
+        ..Default::default()
+    };
+
+    let meta2 = RelationIndexer
+        .index_resource(dir2.path(), &RelationConfig::default(), &resource)?
+        .unwrap();
+
+    assert_eq!(meta2.records, 1);
+
+    let dir_merge = TempDir::new().unwrap();
+    let merged_meta = RelationIndexer.merge(
+        dir_merge.path(),
+        RelationConfig::default(),
+        TestOpener::new(vec![(meta1.clone(), 1u64.into()), (meta2.clone(), 2u64.into())], vec![]),
+    )?;
+
+    assert_eq!(merged_meta.records, 2);
+
+    let dir_merge2 = TempDir::new().unwrap();
+    let merged_meta2 = RelationIndexer.merge(
+        dir_merge2.path(),
+        RelationConfig::default(),
+        TestOpener::new(
+            vec![(meta1, 1u64.into()), (meta2, 2u64.into())],
+            vec![("01808bbd8e784552967a4fb0d8b6e584".to_string(), 3u64.into())],
+        ),
+    )?;
+
+    assert_eq!(merged_meta2.records, 1);
+
+    Ok(())
+}

{nidx_binding-6.9.7.post702 → nidx_binding-6.9.7.post704}/pyproject.toml

@@ -13,7 +13,7 @@ classifiers = [
     "Programming Language :: Python :: Implementation :: CPython",
     "Programming Language :: Python :: Implementation :: PyPy",
 ]
-version = "6.9.7.post702"
+version = "6.9.7.post704"
 
 [project.urls]
 Homepage = "https://nuclia.com"