Spaces:
Build error
Build error
File size: 2,344 Bytes
84d2a97 |
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 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
#[allow(unused)]
mod metrics;
#[cfg(test)]
mod tests {
use quantization::encoded_vectors::{DistanceType, EncodedVectors, VectorParameters};
use quantization::encoded_vectors_u8::EncodedVectorsU8;
use quantization::EncodedVectorsPQ;
use tempfile::Builder;
#[test]
fn empty_data_u8() {
let dir = Builder::new().prefix("storage_dir").tempdir().unwrap();
let vectors_count = 0;
let vector_dim = 256;
let vector_parameters = VectorParameters {
dim: vector_dim,
count: vectors_count,
distance_type: DistanceType::Dot,
invert: false,
};
let vector_data: Vec<Vec<f32>> = Default::default();
let encoded = EncodedVectorsU8::encode(
vector_data.iter(),
Vec::<u8>::new(),
&vector_parameters,
None,
|| false,
)
.unwrap();
let data_path = dir.path().join("data.bin");
let meta_path = dir.path().join("meta.json");
encoded
.save(data_path.as_path(), meta_path.as_path())
.unwrap();
EncodedVectorsU8::<Vec<u8>>::load(
data_path.as_path(),
meta_path.as_path(),
&vector_parameters,
)
.unwrap();
}
#[test]
fn empty_data_pq() {
let dir = Builder::new().prefix("storage_dir").tempdir().unwrap();
let vectors_count = 0;
let vector_dim = 8;
let vector_parameters = VectorParameters {
dim: vector_dim,
count: vectors_count,
distance_type: DistanceType::Dot,
invert: false,
};
let vector_data: Vec<Vec<f32>> = Default::default();
let encoded = EncodedVectorsPQ::encode(
vector_data.iter(),
Vec::<u8>::new(),
&vector_parameters,
2,
1,
|| false,
)
.unwrap();
let data_path = dir.path().join("data.bin");
let meta_path = dir.path().join("meta.json");
encoded
.save(data_path.as_path(), meta_path.as_path())
.unwrap();
EncodedVectorsPQ::<Vec<u8>>::load(
data_path.as_path(),
meta_path.as_path(),
&vector_parameters,
)
.unwrap();
}
}
|