net.jqwik.vavr.arbitraries.collection.queue.VavrPriorityQueueArbitrary Maven / Gradle / Ivy

Go to download

package net.jqwik.vavr.arbitraries.collection.queue;

import net.jqwik.vavr.NaturalComparator;
import net.jqwik.vavr.arbitraries.base.ListBasedArbitrary;
import io.vavr.collection.PriorityQueue;
import net.jqwik.api.Arbitrary;

import java.util.Comparator;

public class VavrPriorityQueueArbitrary extends ListBasedArbitrary> {

	private final Comparator comparator;

	public VavrPriorityQueueArbitrary(
			final Arbitrary elementArbitrary,
			final Comparator comparator) {
		super(elementArbitrary);
		this.comparator = comparator;
	}

	public VavrPriorityQueueArbitrary(final Arbitrary elementArbitrary) {
		this(elementArbitrary, NaturalComparator.instance());
	}

	@Override
	protected PriorityQueue convertJavaListToVavrCollection(final java.util.List javaList) {
		return PriorityQueue.ofAll(this.comparator, javaList);
	}

}