NaiveBoundedRangeNumberGeneratorTest.java

package com.ldbc.driver.generator;

import com.google.common.collect.Range;
import com.ldbc.driver.util.Bucket;
import com.ldbc.driver.util.Histogram;

import java.util.Iterator;

public class NaiveBoundedRangeNumberGeneratorTest extends NumberGeneratorTest<Long, Long> {
    private final long min = 0;
    private final long max = 5;
    private final long boundedMin = 1;
    private final long boundedMax = 3;

    @Override
    public double getMeanTolerance() {
        return 0.1;
    }

    @Override
    public double getDistributionTolerance() {
        return 0.01;
    }

    @Override
    public Iterator<Long> getGeneratorImpl(GeneratorFactory generatorFactory) {
        return generatorFactory.naiveBoundedNumberRange(boundedMin, boundedMax, generatorFactory.uniform(min, max));
    }

    @Override
    public Histogram<Long, Long> getExpectedDistribution() {
        Histogram<Long, Long> expectedDistribution = new Histogram<>(0l);
        expectedDistribution.addBucket(new Bucket.NumberRangeBucket<Long>(Range.closedOpen(-0.5d, 0.5d)), 0l);
        expectedDistribution.addBucket(new Bucket.NumberRangeBucket<Long>(Range.closedOpen(0.5d, 1.5d)), 1l);
        expectedDistribution.addBucket(new Bucket.NumberRangeBucket<Long>(Range.closedOpen(1.5d, 2.5d)), 1l);
        expectedDistribution.addBucket(new Bucket.NumberRangeBucket<Long>(Range.closedOpen(2.5d, 3.5d)), 1l);
        expectedDistribution.addBucket(new Bucket.NumberRangeBucket<Long>(Range.closedOpen(3.5d, 4.5d)), 0l);
        expectedDistribution.addBucket(new Bucket.NumberRangeBucket<Long>(Range.closed(4.5d, 5.5d)), 0l);
        return expectedDistribution;
    }

    @Override
    public double getExpectedMean() {
        return boundedMin + (((double) boundedMax - (double) boundedMin) / 2);
    }

}