@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 } }