예제 #1
0
  @Test
  public void testTopItemItem() throws Exception {
    List<GenericItemSimilarity.ItemItemSimilarity> sims = Lists.newArrayList();
    for (int i = 0; i < 99; i++) {
      sims.add(new GenericItemSimilarity.ItemItemSimilarity(i, i + 1, i / 99.0));
    }

    List<GenericItemSimilarity.ItemItemSimilarity> res =
        TopItems.getTopItemItemSimilarities(10, sims.iterator());
    int gold = 99;
    for (GenericItemSimilarity.ItemItemSimilarity re : res) {
      assertEquals(gold--, re.getItemID2()); // the second id should be equal to 99 to start
    }
  }