pub fn get_mem_index_file(index_path_prefix: &str) -> String {
index_path_prefix.to_string() + "_mem.index"
}
pub fn get_mem_index_data_file(mem_index_path: &str) -> String {
format!("{}.data", mem_index_path)
}
pub fn get_disk_index_file(index_path_prefix: &str) -> String {
index_path_prefix.to_string() + "_disk.index"
}
pub fn get_pq_pivot_file(index_path_prefix: &str) -> String {
index_path_prefix.to_string() + "_pq_pivots.bin"
}
pub fn get_compressed_pq_file(index_path_prefix: &str) -> String {
index_path_prefix.to_string() + "_pq_compressed.bin"
}
pub fn get_disk_index_pq_pivot_file(index_path_prefix: &str) -> String {
index_path_prefix.to_string() + "_disk.index_pq_pivots.bin"
}
pub fn get_disk_index_compressed_pq_file(index_path_prefix: &str) -> String {
index_path_prefix.to_string() + "_disk.index_pq_compressed.bin"
}
pub fn get_label_file(index_path_prefix: &str) -> String {
index_path_prefix.to_string() + "_labels.txt"
}
pub fn get_label_medoids_file(index_path_prefix: &str) -> String {
index_path_prefix.to_string() + "_labels_to_medoids.txt"
}
pub fn get_universal_label_file(index_path_prefix: &str) -> String {
index_path_prefix.to_string() + "_universal_label.txt"
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_get_label_file() {
let prefix = "test_prefix";
let result = get_label_file(prefix);
assert_eq!(result, "test_prefix_labels.txt");
}
#[test]
fn test_get_label_medoids_file() {
let prefix = "test_prefix";
let result = get_label_medoids_file(prefix);
assert_eq!(result, "test_prefix_labels_to_medoids.txt");
}
#[test]
fn test_get_universal_label_file() {
let prefix = "test_prefix";
let result = get_universal_label_file(prefix);
assert_eq!(result, "test_prefix_universal_label.txt");
}
}