TestBenchmarkFunctions

Complexity

Total Complexity

10

Size/Duplication

Total Lines	52
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

10 Methods

Rating	Name	Duplication	Size	Complexity
A	test_schwefel()	0	4	1
A	setUp()	0	5	1
A	test_ackley()	0	4	1
A	test_sphere()	0	4	1
A	test_griewank()	0	4	1
A	test_rastrigin()	0	4	1
A	test_rosenbrock()	0	4	1
A	test_schwefel221()	0	4	1
A	test_whitley()	0	4	1
A	test_schwefel222()	0	4	1

import unittest

import sys
sys.path.append('../')

from NiaPy.benchmarks.rastrigin import Rastrigin
from NiaPy.benchmarks.rosenbrock import Rosenbrock
from NiaPy.benchmarks.griewank import Griewank
from NiaPy.benchmarks.sphere import Sphere
from NiaPy.benchmarks.ackley import Ackley
from NiaPy.benchmarks.schwefel import Schwefel
from NiaPy.benchmarks.schwefel import Schwefel221
from NiaPy.benchmarks.schwefel import Schwefel222
from NiaPy.benchmarks.whitley import Whitley
from NiaPy.benchmarks.utility import Utility


class TestBenchmarkFunctions(unittest.TestCase):

    def setUp(self):
        self.D = 5
        self.array = [0, 0, 0, 0, 0]
        self.array2 = [1, 1, 1, 1, 1]
        self.array3 = [420.9687, 420.9687, 420.9687, 420.9687, 420.9687]

    def test_rastrigin(self):
        rastrigin = Utility().get_benchmark('rastrigin')
        fun = rastrigin.function()
        self.assertEquals(fun(self.D, self.array), 0.0)

    def test_rosenbrock(self):
        rosenbrock = Utility().get_benchmark('rosenbrock')
        fun = rosenbrock.function()
        self.assertEquals(fun(self.D, self.array2), 0.0)

    def test_griewank(self):
        griewank = Utility().get_benchmark('griewank')
        fun = griewank.function()
        self.assertEquals(fun(self.D, self.array), 0.0)

    def test_sphere(self):
        sphere = Utility().get_benchmark('sphere')
        fun = sphere.function()
        self.assertEquals(fun(self.D, self.array), 0.0)

    def test_ackley(self):
        ackley = Utility().get_benchmark('ackley')
        fun = ackley.function()
        self.assertEquals(round(fun(self.D, self.array)), 0.0)

    def test_schwefel(self):
        schwefel = Utility().get_benchmark('schwefel')
        fun = schwefel.function()
        self.assertAlmostEquals(round(fun(self.D, self.array3)), 0.0)

    def test_schwefel221(self):
        schwefel221 = Utility().get_benchmark('schwefel221')
        fun = schwefel221.function()
        self.assertEquals(fun(self.D, self.array), 0.0)

    def test_schwefel222(self):
        schwefel222 = Utility().get_benchmark('schwefel222')
        fun = schwefel222.function()
        self.assertEquals(fun(self.D, self.array), 0.0)

    def test_whitley(self):
        whitley = Utility().get_benchmark('whitley')
        fun = whitley.function()
        self.assertEquals(fun(self.D, self.array2), 0.0)


if __name__ == '__main__':
    unittest.main()