Utility

Complexity

Total Complexity

6

Size/Duplication

Total Lines	17
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	1

1 Method

Rating	Name	Duplication	Size	Complexity
B	itialize_benchmark()	0	15	6

"""Utilities for benchmarks."""

from . import Rastrigin, Rosenbrock, Griewank, Sphere

__all__ = ['Utility']


class Utility(object):

    @staticmethod
    def itialize_benchmark(function):
        if callable(function):
            return function
        else:
            if function == 'rastrigin':
                return Rastrigin.function()
            elif function == 'rosenbrock':
                return Rosenbrock.function()
            elif function == 'griewank':
                return Griewank.function()
            elif function == 'sphere':
                return Sphere.function()
            else:
                raise TypeError('Passed benchmark is not defined!')