Example usage for org.apache.mahout.cf.taste.impl.common LongPrimitiveIterator peek

Introduction

In this page you can find the example usage for org.apache.mahout.cf.taste.impl.common LongPrimitiveIterator peek.

Prototype

long peek();

Source Link

Usage

From source file:org.easyrec.mahout.EasyrecInMemoryDataModelTest.java

License:Open Source License

@Test
public void testEasyrecDataModel_getItemIDs() throws TasteException {
    DataModel easyrecDataModel = new EasyrecDataModel(TENANT_ID, RATE_ACTION_TYPE_ID, HAS_RATING_VALUES,
            mahoutDataModelMappingDAO);/*from  ww w.ja  v a 2 s . c o  m*/
    easyrecDataModel = new EasyrecInMemoryDataModel(easyrecDataModel);
    LongPrimitiveIterator itemIds = easyrecDataModel.getItemIDs();

    String ids = "";
    while (itemIds.hasNext()) {
        ids += itemIds.next();
    }

    assertEquals("123", ids);

    //Test peek function of the iterator
    itemIds = easyrecDataModel.getItemIDs();

    ids = "";
    ids += itemIds.peek();

    while (itemIds.hasNext()) {
        ids += itemIds.peek() + "-" + itemIds.next();
        if (itemIds.hasNext()) {
            ids += "-" + itemIds.peek() + "-" + itemIds.peek() + "#";
        } else {
            ids += "#";
        }
    }
    assertEquals("11-1-2-2#2-2-3-3#3-3#", ids);
}

---