def is_test_job_enabled()

in src/start_testbuilds.py [0:0]

• 22 lines of code
• 21 McCabe index (conditional complexity)

def is_test_job_enabled(test_type):
    """
    Check to see if a test job is enabled
    See if we should run the tests based on test types and config options.
    """
    if test_type == constants.SAGEMAKER_REMOTE_TESTS and config.is_sm_remote_test_enabled():
        return True
    if test_type == constants.SAGEMAKER_EFA_TESTS and config.is_sm_efa_test_enabled():
        return True
    if test_type == constants.SAGEMAKER_RC_TESTS and config.is_sm_rc_test_enabled():
        return True
    if test_type == constants.SAGEMAKER_BENCHMARK_TESTS and config.is_sm_benchmark_test_enabled():
        return True
    if test_type == constants.EC2_TESTS and config.is_ec2_test_enabled():
        return True
    if test_type == constants.EC2_BENCHMARK_TESTS and config.is_ec2_benchmark_test_enabled():
        return True
    if test_type == constants.ECS_TESTS and config.is_ecs_test_enabled():
        return True
    if test_type == constants.EKS_TESTS and config.is_eks_test_enabled():
        return True
    if test_type == constants.SANITY_TESTS and config.is_sanity_test_enabled():
        return True
    if test_type == constants.SECURITY_TESTS and config.is_security_test_enabled():
        return True

    return False