@Test
public void testDefaultGpuInstanceAssignment() {

    Pair<V1Affinity, Map<String, String>> affinityWithAnnotations = factory.buildV1Affinity(
            newGpuJob(Collections.emptyMap()), JobGenerator.oneBatchTask()
    );
    V1NodeSelector nodeSelector = affinityWithAnnotations.getLeft().getNodeAffinity().getRequiredDuringSchedulingIgnoredDuringExecution();
    assertThat(nodeSelector.getNodeSelectorTerms()).hasSize(1);
    assertThat(nodeSelector.getNodeSelectorTerms().get(0).getMatchExpressions().get(0).getValues().get(0)).isEqualTo(DEFAULT_GPU_INSTANCE_TYPE);
}