Merge pull request #380 from TaroKingCN/main

add test_enes.py

pypop7/optimizers/nes/test_enes.py

@@ -0,0 +1,16 @@
+def test_optimize():
+    import numpy  # engine for numerical computing
+    from pypop7.benchmarks.base_functions import rosenbrock  # function to be minimized
+    from pypop7.optimizers.nes.enes import ENES
+    problem = {'fitness_function': rosenbrock,  # to define problem arguments
+               'ndim_problem': 2,
+               'lower_boundary': -5.0 * numpy.ones((2,)),
+               'upper_boundary': 5.0 * numpy.ones((2,))}
+    options = {'max_function_evaluations': 5000,  # to set optimizer options
+               'seed_rng': 2022,
+               'mean': 3.0 * numpy.ones((2,)),
+               'sigma': 3.0}  # global step-size may need to be tuned for optimality
+    enes = ENES(problem, options)  # to initialize the optimizer class
+    results = enes.optimize()  # to run the optimization process
+    assert results['n_function_evaluations'] == 5000
+    assert results['best_so_far_y'] < 1.0