@Test
public void testGpuInstanceAssignment() {

    List<V1Toleration> tolerations = factory.buildV1Toleration(newGpuJob(), JobGenerator.oneBatchTask());
    V1Toleration gpuToleration = tolerations.stream().filter(t -> t.getKey().equals(KubeConstants.TAINT_GPU_INSTANCE)).findFirst().orElse(null);
    assertThat(gpuToleration).isEqualTo(Tolerations.TOLERATION_GPU_INSTANCE);
}