Add test_random_search

Author: skoch9

src/pso.cpp

@@ -24,8 +24,9 @@ population  number of particles in swarm
 
 Returns
 -------
+f(X) objective corresponding to best particle seen so far
 X  best particle seen so far
-Returns objective corresponding to best particle seen so far
+
 
 See also
 --------

src/random_search.cpp

@@ -1,6 +1,6 @@
 #include <npe.h>
 #include <typedefs.h>
-
+#include <pybind11/functional.h>
 #include <pybind11/stl.h>
 
 
@@ -25,8 +25,8 @@ iters  number of iterations
 
 Returns
 -------
+f(X)
 X  #X optimal parameter vector
-Returns f(X)
 
 See also
 --------
@@ -57,4 +57,3 @@ npe_begin_code()
   return std::make_tuple(obj, npe::move(x));
 
 npe_end_code()
-

tests/test_basic.py

@@ -2045,6 +2045,21 @@ def banana(x):
         self.assertTrue(xopt.dtype == lb.dtype)
         self.assertTrue(xopt.shape == (2, ))
 
+    def test_random_search(self):
+        def banana(x):
+            x1 = x[0]
+            x2 = x[1]
+            return x1**4 - 2*x2*x1**2 + x2**2 + x1**2 - 2*x1 + 5
+
+        lb = np.array([-3.0, -1.0])
+        ub = np.array([2.0, 6.0])
+
+        fopt, xopt = igl.random_search(banana, lb, ub, iters=10)
+        
+        self.assertTrue(xopt.flags.c_contiguous)
+        self.assertTrue(xopt.dtype == lb.dtype)
+        self.assertTrue(xopt.shape == (2, ))        
+
 
 if __name__ == '__main__':
     unittest.main()