Search in sources :

Example 1 with ResourceType

use of org.apache.hadoop.yarn.server.resourcemanager.resource.ResourceType in project hadoop by apache.

the class TestDominantResourceFairnessPolicy method testCalculateShares.

@Test
public void testCalculateShares() {
    Resource used = Resources.createResource(10, 5);
    Resource capacity = Resources.createResource(100, 10);
    ResourceType[] resourceOrder = new ResourceType[2];
    ResourceWeights shares = new ResourceWeights();
    DominantResourceFairnessPolicy.DominantResourceFairnessComparator comparator = new DominantResourceFairnessPolicy.DominantResourceFairnessComparator();
    comparator.calculateShares(used, capacity, shares, resourceOrder, ResourceWeights.NEUTRAL);
    assertEquals(.1, shares.getWeight(ResourceType.MEMORY), .00001);
    assertEquals(.5, shares.getWeight(ResourceType.CPU), .00001);
    assertEquals(ResourceType.CPU, resourceOrder[0]);
    assertEquals(ResourceType.MEMORY, resourceOrder[1]);
}
Also used : Resource(org.apache.hadoop.yarn.api.records.Resource) ResourceType(org.apache.hadoop.yarn.server.resourcemanager.resource.ResourceType) ResourceWeights(org.apache.hadoop.yarn.server.resourcemanager.resource.ResourceWeights) Test(org.junit.Test)

Aggregations

Resource (org.apache.hadoop.yarn.api.records.Resource)1 ResourceType (org.apache.hadoop.yarn.server.resourcemanager.resource.ResourceType)1 ResourceWeights (org.apache.hadoop.yarn.server.resourcemanager.resource.ResourceWeights)1 Test (org.junit.Test)1