def test_rosenbrock_output(common_minima2):
"""Tests rosenbrock function output."""
assert np.array_equal(
fx.rosenbrock_func(common_minima2).all(),
np.zeros(3).all())
def test_rosenbrock_output(self):
"""Tests rosenbrock function output."""
self.assertEqual(fx.rosenbrock_func(self.input2).all(),np.zeros(3).all())
def test_rosenbrock_output_size(self):
"""Tests rosenbrock output size."""
self.assertEqual(fx.rosenbrock_func(self.input).shape, self.target_size)
def test_rosenbrock_output_size(common_minima, targetdim):
"""Tests rosenbrock output size."""
assert fx.rosenbrock_func(common_minima).shape == targetdim