net.jqwik.engine.properties.arbitraries.randomized.BiasedNumericGenerator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jqwik-engine Show documentation

Jqwik Engine module

There is a newer version: 1.9.1

package net.jqwik.engine.properties.arbitraries.randomized;

import java.math.*;
import java.util.*;

import net.jqwik.api.*;
import net.jqwik.api.RandomDistribution.*;
import net.jqwik.engine.properties.*;

class BiasedNumericGenerator implements RandomNumericGenerator {

	private final RandomNumericGenerator partitionedGenerator;

	BiasedNumericGenerator(int genSize, BigInteger min, BigInteger max, BigInteger center) {
		List partitionPoints = BiasedPartitionPointsCalculator.calculatePartitionPoints(genSize, min, max, center);
		Range range = Range.of(min, max);
		partitionedGenerator = partitionedGenerator(range, partitionPoints);
	}

	@Override
	public BigInteger next(Random random) {
		return partitionedGenerator.next(random);
	}

	private RandomNumericGenerator partitionedGenerator(
		Range range,
		List partitionPoints
	) {
		if (partitionPoints.isEmpty()) {
			return createUniformGenerator(range.min, range.max);
		}
		List generators = createPartitions(range, partitionPoints);
		return random -> generators.get(random.nextInt(generators.size())).next(random);
	}

	private List createPartitions(
		Range range,
		List partitionPoints
	) {
		List partitions = new ArrayList<>();
		Collections.sort(partitionPoints);
		BigInteger lower = range.min;
		for (BigInteger partitionPoint : partitionPoints) {
			BigInteger upper = partitionPoint;
			if (upper.compareTo(lower) <= 0) {
				continue;
			}
			if (upper.compareTo(range.max) >= 0) {
				break;
			}
			partitions.add(createUniformGenerator(lower, upper.subtract(BigInteger.ONE)));
			lower = upper;
		}
		partitions.add(createUniformGenerator(lower, range.max));
		return partitions;
	}

	private RandomNumericGenerator createUniformGenerator(
		BigInteger minGenerate,
		BigInteger maxGenerate
	) {
		int ignoredGenSize = 1000;
		BigInteger ignoredCenter = minGenerate;
		return RandomDistribution.uniform().createGenerator(ignoredGenSize, minGenerate, maxGenerate, ignoredCenter);
	}

}