How to use

findTop

method

in

net.ricecode.similarity.StringSimilarityService

@Test 
public void testFindTop() {
  SimilarityStrategy strategy = mock(SimilarityStrategy.class);
  String target = "McDonalds";
  String c1 = "MacMahons";
  String c2 = "McPherson";
  String c3 = "McDonalds";
  
  SimilarityScore expected = new SimilarityScore(c3, 1.000);
  
  when(strategy.score(c1, target)).thenReturn(0.90);
  when(strategy.score(c2, target)).thenReturn(0.74);
  when(strategy.score(c3, target)).thenReturn(1.000);
  
  StringSimilarityService service = new StringSimilarityServiceImpl(strategy);
  List<String> features = new ArrayList<String>();
  features.add(c1);
  features.add(c2);
  features.add(c3);
  
  SimilarityScore top= service.findTop(features,target);
  verify(strategy).score(c1, target);
  verify(strategy).score(c2, target);
  verify(strategy).score(c3, target);
  assertEquals(expected, top);
  
}

@Test 
public void testFindTop_Ascending() {
  SimilarityStrategy strategy = mock(SimilarityStrategy.class);
  String target = "McDonalds";
  String c1 = "MacMahons";
  String c2 = "McPherson";
  String c3 = "McDonalds";
  
  SimilarityScore expected = new SimilarityScore(c2, 0.74);
  
  when(strategy.score(c1, target)).thenReturn(0.90);
  when(strategy.score(c2, target)).thenReturn(0.74);
  when(strategy.score(c3, target)).thenReturn(1.000);
  
  StringSimilarityService service = new StringSimilarityServiceImpl(strategy);
  List<String> features = new ArrayList<String>();
  features.add(c1);
  features.add(c2);
  features.add(c3);
  
  AscendingSimilarityScoreComparator comparator = new AscendingSimilarityScoreComparator();
  SimilarityScore top= service.findTop(features,target, comparator);
  verify(strategy).score(c1, target);
  verify(strategy).score(c2, target);
  verify(strategy).score(c3, target);
  assertEquals(expected, top);
}

  @Test 
  public void testFindTop_Descending() {
    SimilarityStrategy strategy = mock(SimilarityStrategy.class);
    String target = "McDonalds";
    String c1 = "MacMahons";
    String c2 = "McPherson";
    String c3 = "McDonalds";
    
    SimilarityScore expected = new SimilarityScore(c3, 1.000);
    
    when(strategy.score(c1, target)).thenReturn(0.90);
    when(strategy.score(c2, target)).thenReturn(0.74);
    when(strategy.score(c3, target)).thenReturn(1.000);
    
    StringSimilarityService service = new StringSimilarityServiceImpl(strategy);
    List<String> features = new ArrayList<String>();
    features.add(c1);
    features.add(c2);
    features.add(c3);
    
    DescendingSimilarityScoreComparator comparator = new DescendingSimilarityScoreComparator();
    SimilarityScore top= service.findTop(features,target, comparator);
    verify(strategy).score(c1, target);
    verify(strategy).score(c2, target);
    verify(strategy).score(c3, target);
    assertEquals(expected, top);
    
  }
}