src/test/java/com/yahoo/sketches/BoundsOnRatiosInThetaSketchedSetsTest.java

/*
 * Copyright 2016, Yahoo! Inc.
 * Licensed under the terms of the Apache License 2.0. See LICENSE file at the project root for terms.
 */

package com.yahoo.sketches;

import static org.testng.Assert.*;

import com.yahoo.sketches.theta.*;
import org.testng.annotations.Test;

public class BoundsOnRatiosInThetaSketchedSetsTest {

  @Test
  public void checkNormalReturns() {
    UpdateSketch skA = Sketches.updateSketchBuilder().build(); //4K
    UpdateSketch skC = Sketches.updateSketchBuilder().build();
    int uA = 10000;
    int uC = 100000;
    for (int i=0; i<uA; i++) { skA.update(i); }
    for (int i=0; i<uC; i++) { skC.update(i+uA/2); }
    Intersection inter = Sketches.setOperationBuilder().buildIntersection();
    inter.update(skA);
    inter.update(skC);
    CompactSketch skB = inter.getResult();

    double est = BoundsOnRatiosInThetaSketchedSets.getEstimateOfBoverA(skA, skB);
    double lb = BoundsOnRatiosInThetaSketchedSets.getLowerBoundForBoverA(skA, skB);
    double ub = BoundsOnRatiosInThetaSketchedSets.getUpperBoundForBoverA(skA, skB);
    assertTrue(ub > est);
    assertTrue(est > lb);
    assertEquals(est, 0.5, .03);
    println("ub : "+ ub);
    println("est: "+est);
    println("lb : "+lb);
    skA.reset(); //skA is now empty
    est = BoundsOnRatiosInThetaSketchedSets.getEstimateOfBoverA(skA, skB);
    lb = BoundsOnRatiosInThetaSketchedSets.getLowerBoundForBoverA(skA, skB);
    ub = BoundsOnRatiosInThetaSketchedSets.getUpperBoundForBoverA(skA, skB);
    println("ub : "+ ub);
    println("est: "+est);
    println("lb : "+lb);
    skC.reset(); //Now both are empty
    est = BoundsOnRatiosInThetaSketchedSets.getEstimateOfBoverA(skA, skC);
    lb = BoundsOnRatiosInThetaSketchedSets.getLowerBoundForBoverA(skA, skC);
    ub = BoundsOnRatiosInThetaSketchedSets.getUpperBoundForBoverA(skA, skC);
    println("ub : "+ ub);
    println("est: "+est);
    println("lb : "+lb);
  }

  @Test(expectedExceptions = IllegalArgumentException.class)
  public void checkAbnormalReturns() {
    UpdateSketch skA = Sketches.updateSketchBuilder().build(); //4K
    UpdateSketch skC = Sketches.updateSketchBuilder().build();
    int uA = 100000;
    int uC = 10000;
    for (int i=0; i<uA; i++) { skA.update(i); }
    for (int i=0; i<uC; i++) { skC.update(i+uA/2); }
    BoundsOnRatiosInThetaSketchedSets.getEstimateOfBoverA(skA, skC);
  }

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