| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| |
| package org.apache.datasketches.hll; |
| |
| import static org.testng.Assert.assertEquals; |
| import static org.testng.Assert.assertNotNull; |
| import static org.testng.Assert.assertNull; |
| import static org.testng.Assert.assertTrue; |
| |
| import org.testng.annotations.Test; |
| |
| import org.apache.datasketches.memory.WritableMemory; |
| |
| /** |
| * @author Lee Rhodes |
| */ |
| @SuppressWarnings("javadoc") |
| public class CouponListTest { |
| |
| @Test |
| public void checkIterator() { |
| checkIterator(false); |
| checkIterator(true); |
| } |
| |
| private static void checkIterator(boolean direct) { |
| int lgConfigK = 8; |
| TgtHllType tgtHllType = TgtHllType.HLL_4; |
| int bytes = HllSketch.getMaxUpdatableSerializationBytes(lgConfigK, tgtHllType); |
| WritableMemory wmem = WritableMemory.allocate(bytes); |
| HllSketch sk = (direct) ? new HllSketch(lgConfigK, tgtHllType, wmem) : new HllSketch(8); |
| for (int i = 0; i < 7; i++) { sk.update(i); } |
| PairIterator itr = sk.iterator(); |
| println(itr.getHeader()); |
| while (itr.nextAll()) { |
| int key = itr.getKey(); |
| int val = itr.getValue(); |
| int idx = itr.getIndex(); |
| int slot = itr.getSlot(); |
| println("Idx: " + idx + ", Key: " + key + ", Val: " + val + ", Slot: " + slot); |
| } |
| } |
| |
| @Test |
| public void checkDuplicatesAndMisc() { |
| checkDuplicatesAndMisc(false); |
| checkDuplicatesAndMisc(true); |
| } |
| |
| private static void checkDuplicatesAndMisc(boolean direct) { |
| int lgConfigK = 8; |
| TgtHllType tgtHllType = TgtHllType.HLL_4; |
| int bytes = HllSketch.getMaxUpdatableSerializationBytes(lgConfigK, tgtHllType); |
| WritableMemory wmem = WritableMemory.allocate(bytes); |
| HllSketch sk = (direct) ? new HllSketch(lgConfigK, tgtHllType, wmem) : new HllSketch(8); |
| |
| for (int i = 1; i <= 7; i++) { |
| sk.update(i); |
| sk.update(i); |
| } |
| assertEquals(sk.getCurMode(), CurMode.LIST); |
| assertEquals(sk.getCompositeEstimate(), 7.0, 7 * .01); |
| if (direct) { |
| assertNotNull(sk.hllSketchImpl.getWritableMemory()); |
| } else { |
| assertNull(sk.hllSketchImpl.getWritableMemory()); |
| } |
| |
| sk.update(8); |
| sk.update(8); |
| assertEquals(sk.getCurMode(), CurMode.SET); |
| assertEquals(sk.getCompositeEstimate(), 8.0, 8 * .01); |
| if (direct) { |
| assertNotNull(sk.hllSketchImpl.getWritableMemory()); |
| } else { |
| assertNull(sk.hllSketchImpl.getWritableMemory()); |
| } |
| |
| for (int i = 9; i <= 25; i++) { |
| sk.update(i); |
| sk.update(i); |
| } |
| assertEquals(sk.getCurMode(), CurMode.HLL); |
| assertEquals(sk.getCompositeEstimate(), 25.0, 25 * .1); |
| if (direct) { |
| assertNotNull(sk.hllSketchImpl.getWritableMemory()); |
| } else { |
| assertNull(sk.hllSketchImpl.getWritableMemory()); |
| } |
| |
| double re = sk.getRelErr(true, true, 4, 1); |
| assertTrue(re < 0.0); |
| } |
| |
| @Test |
| public void toByteArray_Heapify() { |
| toByteArrayHeapify(7); |
| toByteArrayHeapify(21); |
| } |
| |
| private static void toByteArrayHeapify(int lgK) { |
| HllSketch sk1 = new HllSketch(lgK); |
| |
| int u = (lgK < 8) ? 7 : ((1 << (lgK - 3))/4) * 3; |
| for (int i = 0; i < u; i++) { |
| sk1.update(i); |
| } |
| double est1 = sk1.getEstimate(); |
| assertEquals(est1, u, u * 100.0E-6); |
| //println("Original\n" + sk1.toString()); |
| |
| byte[] byteArray = sk1.toCompactByteArray(); |
| //println("Preamble: " + PreambleUtil.toString(byteArray)); |
| HllSketch sk2 = HllSketch.heapify(byteArray); |
| double est2 = sk2.getEstimate(); |
| //println("Heapify Compact\n" + sk2.toString(true, true, true, true)); |
| assertEquals(est2, est1, 0.0); |
| |
| byteArray = sk1.toUpdatableByteArray(); |
| sk2 = HllSketch.heapify(byteArray); |
| est2 = sk2.getEstimate(); |
| //println("Heapify Updatable\n" + sk2.toString()); |
| assertEquals(est2, est1, 0.0); |
| } |
| |
| @Test |
| public void checkGetMemory() { |
| HllSketch sk1 = new HllSketch(4); |
| AbstractCoupons absCoup = (AbstractCoupons) sk1.hllSketchImpl; |
| assertNull(absCoup.getMemory()); |
| } |
| |
| @Test |
| public void printlnTest() { |
| println("PRINTING: "+this.getClass().getName()); |
| } |
| |
| /** |
| * @param s value to print |
| */ |
| static void println(String s) { |
| //System.out.println(s); //disable here |
| } |
| } |