distance_tests.rs

extern crate natural;

use natural::distance::jaro_winkler_distance;
use natural::distance::levenshtein_distance;

#[test]
fn test_jaro_winkler() {
  f64_eq(jaro_winkler_distance("one", "one"), 1f32);
  f64_eq(jaro_winkler_distance("one", ""), 0f32);
  f64_eq(jaro_winkler_distance("", ""), 0f32);
  f64_eq(jaro_winkler_distance("", "one"), 0f32);
  f64_eq(jaro_winkler_distance("DWAYNE", "DUANE"), 0.822);
  f64_eq(jaro_winkler_distance("Martha", "Marhta"), 0.944);
  f64_eq(jaro_winkler_distance("dicksonx", "dixon"), 0.767);
}

fn f64_eq(a: f32, b: f32) {
  assert!((a - b).abs() < 0.01);
}

// TODO: test lev distance

#[test]
fn test_lev_distance() {
  assert_eq!(levenshtein_distance("", ""), 0);
  assert_eq!(levenshtein_distance("one", "one"), 0);
  assert_eq!(levenshtein_distance("one", ""), 3);
  assert_eq!(levenshtein_distance("", "one"), 3);
  assert_eq!(levenshtein_distance("kitten", "sitting"), 3);
  assert_eq!(levenshtein_distance("sitting", "kitten"), 3);
}