ResourceCalculator { public abstract Resource normalize(Resource r, Resource minimumResource, Resource maximumResource, Resource stepFactor);  abstract int compare(
      Resource clusterResource, Resource lhs, Resource rhs, boolean singleType); int compare(Resource clusterResource, Resource lhs, Resource rhs); static int divideAndCeil(int a, int b); static int divideAndCeil(int a, float b); static long divideAndCeil(long a, long b); static long divideAndCeil(long a, float b); static int roundUp(int a, int b); static long roundUp(long a, long b); static long roundDown(long a, long b); static int roundDown(int a, int b); abstract long computeAvailableContainers(
      Resource available, Resource required); abstract Resource multiplyAndNormalizeUp(
      Resource r, double by, Resource stepFactor); abstract Resource multiplyAndNormalizeUp(
      Resource r, double[] by, Resource stepFactor); abstract Resource multiplyAndNormalizeDown(
      Resource r, double by, Resource stepFactor); abstract Resource normalize(Resource r, Resource minimumResource,
      Resource maximumResource, Resource stepFactor); abstract Resource roundUp(Resource r, Resource stepFactor); abstract Resource roundDown(Resource r, Resource stepFactor); abstract float divide(
      Resource clusterResource, Resource numerator, Resource denominator); abstract boolean isInvalidDivisor(Resource r); abstract float ratio(Resource a, Resource b); abstract Resource divideAndCeil(Resource numerator, int denominator); abstract Resource divideAndCeil(Resource numerator, float denominator); abstract boolean fitsIn(Resource smaller, Resource bigger); abstract boolean isAnyMajorResourceZeroOrNegative(Resource resource); abstract Resource normalizeDown(Resource r, Resource stepFactor); abstract boolean isAnyMajorResourceAboveZero(Resource resource);  }
@Test(timeout = 10000) public void testNormalize() { Resource ask = Resource.newInstance(1111, 2); Resource min = Resource.newInstance(1024, 1); Resource max = Resource.newInstance(8 * 1024, 8); Resource increment = Resource.newInstance(1024, 4); if (resourceCalculator instanceof DefaultResourceCalculator) { Resource result = Resources.normalize(resourceCalculator, ask, min, max, increment); assertEquals(2 * 1024, result.getMemorySize()); } else if (resourceCalculator instanceof DominantResourceCalculator) { Resource result = Resources.normalize(resourceCalculator, ask, min, max, increment); assertEquals(2 * 1024, result.getMemorySize()); assertEquals(4, result.getVirtualCores()); } ask = Resource.newInstance(512, 0); min = Resource.newInstance(2 * 1024, 2); max = Resource.newInstance(8 * 1024, 8); increment = Resource.newInstance(1024, 1); if (resourceCalculator instanceof DefaultResourceCalculator) { Resource result = Resources.normalize(resourceCalculator, ask, min, max, increment); assertEquals(2 * 1024, result.getMemorySize()); } else if (resourceCalculator instanceof DominantResourceCalculator) { Resource result = Resources.normalize(resourceCalculator, ask, min, max, increment); assertEquals(2 * 1024, result.getMemorySize()); assertEquals(2, result.getVirtualCores()); } ask = Resource.newInstance(9 * 1024, 9); min = Resource.newInstance(2 * 1024, 2); max = Resource.newInstance(8 * 1024, 8); increment = Resource.newInstance(1024, 1); if (resourceCalculator instanceof DefaultResourceCalculator) { Resource result = Resources.normalize(resourceCalculator, ask, min, max, increment); assertEquals(8 * 1024, result.getMemorySize()); } else if (resourceCalculator instanceof DominantResourceCalculator) { Resource result = Resources.normalize(resourceCalculator, ask, min, max, increment); assertEquals(8 * 1024, result.getMemorySize()); assertEquals(8, result.getVirtualCores()); } ask = Resource.newInstance(1111, 2); min = Resource.newInstance(2 * 1024, 2); max = Resource.newInstance(8 * 1024, 8); increment = Resource.newInstance(0, 0); if (resourceCalculator instanceof DefaultResourceCalculator) { Resource result = Resources.normalize(resourceCalculator, ask, min, max, increment); assertEquals(2 * 1024, result.getMemorySize()); } else if (resourceCalculator instanceof DominantResourceCalculator) { Resource result = Resources.normalize(resourceCalculator, ask, min, max, increment); assertEquals(2 * 1024, result.getMemorySize()); assertEquals(2, result.getVirtualCores()); } }