mirror of
https://github.com/meilisearch/MeiliSearch
synced 2024-11-11 07:28:56 +01:00
Add snapshot tests for Facets::execute
This commit is contained in:
parent
b9907997e4
commit
3a734af159
7
.gitignore
vendored
7
.gitignore
vendored
@ -6,3 +6,10 @@
|
|||||||
*.csv
|
*.csv
|
||||||
*.mmdb
|
*.mmdb
|
||||||
*.svg
|
*.svg
|
||||||
|
|
||||||
|
# Snapshots
|
||||||
|
## ... large
|
||||||
|
*.full.snap
|
||||||
|
|
||||||
|
# ... unreviewed
|
||||||
|
*.snap.new
|
||||||
|
@ -190,6 +190,17 @@ pub fn documents_batch_reader_from_objects(
|
|||||||
DocumentsBatchReader::from_reader(std::io::Cursor::new(builder.into_inner().unwrap())).unwrap()
|
DocumentsBatchReader::from_reader(std::io::Cursor::new(builder.into_inner().unwrap())).unwrap()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
pub fn batch_reader_from_documents(
|
||||||
|
documents: &[Object],
|
||||||
|
) -> DocumentsBatchReader<std::io::Cursor<Vec<u8>>> {
|
||||||
|
let mut builder = DocumentsBatchBuilder::new(Vec::new());
|
||||||
|
for object in documents {
|
||||||
|
builder.append_json_object(&object).unwrap();
|
||||||
|
}
|
||||||
|
DocumentsBatchReader::from_reader(std::io::Cursor::new(builder.into_inner().unwrap())).unwrap()
|
||||||
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod test {
|
mod test {
|
||||||
use std::io::Cursor;
|
use std::io::Cursor;
|
||||||
|
@ -30,7 +30,7 @@ pub fn default_db_snapshot_settings_for_test(name: Option<&str>) -> insta::Setti
|
|||||||
|
|
||||||
#[macro_export]
|
#[macro_export]
|
||||||
macro_rules! db_snap {
|
macro_rules! db_snap {
|
||||||
($index:ident, $db_name:ident, $name:literal) => {
|
($index:ident, $db_name:ident, $name:expr) => {
|
||||||
let settings = $crate::snapshot_tests::default_db_snapshot_settings_for_test(Some(
|
let settings = $crate::snapshot_tests::default_db_snapshot_settings_for_test(Some(
|
||||||
&format!("{}", $name),
|
&format!("{}", $name),
|
||||||
));
|
));
|
||||||
|
@ -342,3 +342,91 @@ fn write_string_entry(
|
|||||||
writer.insert(&key, &data)?;
|
writer.insert(&key, &data)?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use std::num::NonZeroUsize;
|
||||||
|
|
||||||
|
use crate::{db_snap, documents::batch_reader_from_documents, index::tests::TempIndex};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_facets_number() {
|
||||||
|
let test =
|
||||||
|
|name: &str, group_size: Option<NonZeroUsize>, min_level_size: Option<NonZeroUsize>| {
|
||||||
|
let mut index = TempIndex::new_with_map_size(4096 * 1000 * 10); // 40MB
|
||||||
|
index.index_documents_config.autogenerate_docids = true;
|
||||||
|
index.index_documents_config.facet_level_group_size = group_size;
|
||||||
|
index.index_documents_config.facet_min_level_size = min_level_size;
|
||||||
|
|
||||||
|
index
|
||||||
|
.update_settings(|settings| {
|
||||||
|
settings.set_filterable_fields(
|
||||||
|
IntoIterator::into_iter(["facet".to_owned(), "facet2".to_owned()])
|
||||||
|
.collect(),
|
||||||
|
);
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let mut documents = vec![];
|
||||||
|
for i in 0..1_000 {
|
||||||
|
documents.push(serde_json::json!({ "facet": i }).as_object().unwrap().clone());
|
||||||
|
}
|
||||||
|
for i in 0..100 {
|
||||||
|
documents.push(serde_json::json!({ "facet2": i }).as_object().unwrap().clone());
|
||||||
|
}
|
||||||
|
let documents = batch_reader_from_documents(&documents);
|
||||||
|
|
||||||
|
index.add_documents(documents).unwrap();
|
||||||
|
|
||||||
|
db_snap!(index, facet_id_f64_docids, name);
|
||||||
|
};
|
||||||
|
|
||||||
|
test("default", None, None);
|
||||||
|
test("tiny_groups_tiny_levels", NonZeroUsize::new(1), NonZeroUsize::new(1));
|
||||||
|
test("small_groups_small_levels", NonZeroUsize::new(2), NonZeroUsize::new(2));
|
||||||
|
test("small_groups_large_levels", NonZeroUsize::new(2), NonZeroUsize::new(128));
|
||||||
|
test("large_groups_small_levels", NonZeroUsize::new(16), NonZeroUsize::new(2));
|
||||||
|
test("large_groups_large_levels", NonZeroUsize::new(16), NonZeroUsize::new(256));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_facets_string() {
|
||||||
|
let test = |name: &str,
|
||||||
|
group_size: Option<NonZeroUsize>,
|
||||||
|
min_level_size: Option<NonZeroUsize>| {
|
||||||
|
let mut index = TempIndex::new_with_map_size(4096 * 1000 * 10); // 40MB
|
||||||
|
index.index_documents_config.autogenerate_docids = true;
|
||||||
|
index.index_documents_config.facet_level_group_size = group_size;
|
||||||
|
index.index_documents_config.facet_min_level_size = min_level_size;
|
||||||
|
|
||||||
|
index
|
||||||
|
.update_settings(|settings| {
|
||||||
|
settings.set_filterable_fields(
|
||||||
|
IntoIterator::into_iter(["facet".to_owned(), "facet2".to_owned()])
|
||||||
|
.collect(),
|
||||||
|
);
|
||||||
|
})
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let mut documents = vec![];
|
||||||
|
for i in 0..100 {
|
||||||
|
documents.push(
|
||||||
|
serde_json::json!({ "facet": format!("s{i:X}") }).as_object().unwrap().clone(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
for i in 0..10 {
|
||||||
|
documents.push(
|
||||||
|
serde_json::json!({ "facet2": format!("s{i:X}") }).as_object().unwrap().clone(),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let documents = batch_reader_from_documents(&documents);
|
||||||
|
|
||||||
|
index.add_documents(documents).unwrap();
|
||||||
|
|
||||||
|
db_snap!(index, facet_id_string_docids, name);
|
||||||
|
};
|
||||||
|
|
||||||
|
test("default", None, None);
|
||||||
|
test("tiny_groups_tiny_levels", NonZeroUsize::new(1), NonZeroUsize::new(1));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -0,0 +1,4 @@
|
|||||||
|
---
|
||||||
|
source: milli/src/update/facets.rs
|
||||||
|
---
|
||||||
|
587899707db2848da3f18399e14ed4d0
|
@ -0,0 +1,4 @@
|
|||||||
|
---
|
||||||
|
source: milli/src/update/facets.rs
|
||||||
|
---
|
||||||
|
02bbf2ca1663cccea0e4c06d5ad06a45
|
@ -0,0 +1,4 @@
|
|||||||
|
---
|
||||||
|
source: milli/src/update/facets.rs
|
||||||
|
---
|
||||||
|
e68ea591e1af3e53e544dff9a1648e88
|
@ -0,0 +1,4 @@
|
|||||||
|
---
|
||||||
|
source: milli/src/update/facets.rs
|
||||||
|
---
|
||||||
|
12a4bb0f5b95d7629c2b9a915150c0cf
|
@ -0,0 +1,4 @@
|
|||||||
|
---
|
||||||
|
source: milli/src/update/facets.rs
|
||||||
|
---
|
||||||
|
6438e94bc7fada13022e0efccdf294e0
|
@ -0,0 +1,4 @@
|
|||||||
|
---
|
||||||
|
source: milli/src/update/facets.rs
|
||||||
|
---
|
||||||
|
5348bbc46b5384455b6a900666d2a502
|
@ -0,0 +1,4 @@
|
|||||||
|
---
|
||||||
|
source: milli/src/update/facets.rs
|
||||||
|
---
|
||||||
|
faddef9eae5f2efacfec51f20f2e8cd6
|
@ -0,0 +1,4 @@
|
|||||||
|
---
|
||||||
|
source: milli/src/update/facets.rs
|
||||||
|
---
|
||||||
|
ddb8fc987c5dc892337682595043858e
|
Loading…
Reference in New Issue
Block a user