Utility

Complexity

Total Complexity

11

Size/Duplication

Total Lines	28
Duplicated Lines	0 %

Importance

Changes	4
Bugs	0	Features	1

1 Method

Rating	Name	Duplication	Size	Complexity
F	get_benchmark()	0	25	11

"""Utilities for benchmarks."""

from . import Rastrigin, Rosenbrock, Griewank, \
    Sphere, Ackley, Schwefel, Schwefel221, \
    Schwefel222, Whitley

__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()
            elif benchmark == 'schwefel221':
                return Schwefel221()
            elif benchmark == 'schwefel222':
                return Schwefel222()
            elif benchmark == 'whitley':
                return Whitley()
            else:
                raise TypeError('Passed benchmark is not defined!')