org.junit.experimental.theories.internal.EnumSupplier Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of virtdata-lib-curves4 Show documentation

Statistical sampling library for use in virtualdataset libraries, based on apache commons math 4

There is a newer version: 2.12.15

package org.junit.experimental.theories.internal;

import java.util.ArrayList;
import java.util.List;

import org.junit.experimental.theories.ParameterSignature;
import org.junit.experimental.theories.ParameterSupplier;
import org.junit.experimental.theories.PotentialAssignment;

public class EnumSupplier extends ParameterSupplier {

    private Class enumType;

    public EnumSupplier(Class enumType) {
        this.enumType = enumType;
    }

    @Override
    public List getValueSources(ParameterSignature sig) {
        Object[] enumValues = enumType.getEnumConstants();
        
        List assignments = new ArrayList();
        for (Object value : enumValues) {
            assignments.add(PotentialAssignment.forValue(value.toString(), value));
        }
        
        return assignments;
    }

}