Utility

Complexity

Total Complexity

8

Size/Duplication

Total Lines	22
Duplicated Lines	0 %

Importance

Changes	4
Bugs	0	Features	1

1 Method

Rating	Name	Duplication	Size	Complexity
C	get_benchmark()	0	19	8

"""Utilities for benchmarks."""

from . import Rastrigin, Rosenbrock, Griewank, Sphere, Ackley, Schwefel

__all__ = ['Utility']


class Utility(object):

    # pylint: disable=too-many-return-statements
    @staticmethod
    def get_benchmark(benchmark):
        if not isinstance(benchmark, ''.__class__):
            return benchmark
        else:
            if benchmark == 'rastrigin':
                return Rastrigin()
            elif benchmark == 'rosenbrock':
                return Rosenbrock()
            elif benchmark == 'griewank':
                return Griewank()
            elif benchmark == 'sphere':
                return Sphere()
            elif benchmark == 'ackley':
                return Ackley()
            elif benchmark == 'schwefel':
                return Schwefel()
            else:
                raise TypeError('Passed benchmark is not defined!')