lucene-java-3.5.0/lucene/contrib/spellchecker/src/test/org/apache/lucene/search/spell/TestLevenshteinDistance.java

   1 package org.apache.lucene.search.spell;
   2
   3 /**
   4  * Licensed to the Apache Software Foundation (ASF) under one or more
   5  * contributor license agreements.  See the NOTICE file distributed with
   6  * this work for additional information regarding copyright ownership.
   7  * The ASF licenses this file to You under the Apache License, Version 2.0
   8  * (the "License"); you may not use this file except in compliance with
   9  * the License.  You may obtain a copy of the License at
  10  *
  11  *     http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  */
  19
  20 import org.apache.lucene.util.LuceneTestCase;
  21
  22 public class TestLevenshteinDistance extends LuceneTestCase {
  23
  24   private StringDistance sd = new LevensteinDistance();
  25
  26   public void testGetDistance() {
  27     float d = sd.getDistance("al", "al");
  28     assertEquals(d,1.0f,0.001);
  29     d = sd.getDistance("martha", "marhta");
  30     assertEquals(d,0.6666,0.001);
  31     d = sd.getDistance("jones", "johnson");
  32     assertEquals(d,0.4285,0.001);
  33     d = sd.getDistance("abcvwxyz", "cabvwxyz");
  34     assertEquals(d,0.75,0.001);
  35     d = sd.getDistance("dwayne", "duane");
  36     assertEquals(d,0.666,0.001);
  37     d = sd.getDistance("dixon", "dicksonx");
  38     assertEquals(d,0.5,0.001);
  39     d = sd.getDistance("six", "ten");
  40     assertEquals(d,0,0.001);
  41     float d1 = sd.getDistance("zac ephron", "zac efron");
  42     float d2 = sd.getDistance("zac ephron", "kai ephron");
  43     assertEquals(d1,d2,0.001);
  44     d1 = sd.getDistance("brittney spears", "britney spears");
  45     d2 = sd.getDistance("brittney spears", "brittney startzman");
  46     assertTrue(d1 > d2);
  47   }
  48
  49   public void testEmpty() throws Exception {
  50     float d = sd.getDistance("", "al");
  51     assertEquals(d,0.0f,0.001);
  52   }
  53
  54 }