extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2024-11-26 10:23:52 +01:00
Code Issues Packages Projects Releases Wiki Activity

did_you_mean: case-insensitive edit distance, tests (#4999)

Browse Source
This commit is contained in:
Reilly Wood 2022-03-27 15:11:56 -07:00 committed by GitHub
parent 911fba8a8a
commit 5586d4a0a0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 30 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
crates/nu-protocol/src/shell_error.rs
View File

@ -360,19 +360,10 @@ impl From<Box<dyn std::error::Error + Send + Sync>> for ShellError {
}
pub fn did_you_mean(possibilities: &[String], tried: &str) -> Option<String> {
// First see if we can find the same string just cased differently
let great_match_index = possibilities
.iter()
.position(|word| word.to_lowercase() == tried.to_lowercase());
if let Some(index) = great_match_index {
return Some(possibilities[index].to_owned());
}
let mut possible_matches: Vec<_> = possibilities
.iter()
.map(|word| {
let edit_distance = levenshtein_distance(word, tried);
let edit_distance = levenshtein_distance(&word.to_lowercase(), &tried.to_lowercase());
(edit_distance, word.to_owned())
})
.collect();
@ -446,3 +437,32 @@ pub fn levenshtein_distance(a: &str, b: &str) -> usize {
result
}
#[cfg(test)]
mod tests {
use super::did_you_mean;
#[test]
fn did_you_mean_works_with_wrong_case() {
let possibilities = &["OS".into(), "PWD".into()];
let actual = did_you_mean(possibilities, "pwd");
let expected = Some(String::from("PWD"));
assert_eq!(actual, expected)
}
#[test]
fn did_you_mean_works_with_typo() {
let possibilities = &["OS".into(), "PWD".into()];
let actual = did_you_mean(possibilities, "PWF");
let expected = Some(String::from("PWD"));
assert_eq!(actual, expected)
}
#[test]
fn did_you_mean_works_with_wrong_case_and_typo() {
let possibilities = &["OS".into(), "PWD".into()];
let actual = did_you_mean(possibilities, "pwf");
let expected = Some(String::from("PWD"));
assert_eq!(actual, expected)
}
}