Add test for acqfuncs

Author: ardunn

.coveragerc

@@ -4,21 +4,22 @@ exclude_lines =
     if __name__ == .__main__.:
 ignore_errors=True
 omit =
-    *dev_scripts*
-    *examples*
-    *.cicleci*
-    *site-packages*
-    *tests*
-    *versioneer.py
-    *_version.py
-    *.md
-    *.gitignore
-    *.txt
-    *setup.py
-    *__init__*
-    *.egg-info*
-    *docs*
-    *docs_rst*
-    tasks.py
-    setup.py
-    *.git*
+    */dev_scripts/*
+    */docs/*
+    */docs_rst/*
+    */examples/*
+    */.cicleci/*
+    */python?.?/*
+    */site-packages/*
+    *tests/*
+    */versioneer.py
+    */_version.py
+    */*.md
+    */*.yaml
+    */*.cfg
+    */*.gitignore
+    */*.txt
+    */setup.py
+    */__init__.py
+    */rocketsled.egg-info/*
+    */.eggs/*

rocketsled/tests/test_task.py

@@ -313,6 +313,33 @@ def test_missioncontrol(self):
         plt.savefig(savepath)
         self.assertTrue(os.path.exists(savepath))
 
+    def test_acqfuncs(self):
+        self.mc.configure(wf_creator=wf_creator_basic,
+                          dimensions=self.dims_basic,
+                          predictor="GaussianProcessRegressor",
+                          acq=None)
+        launchpad.add_wf(wf_creator_basic([5, 11, 'blue']))
+        rapidfire(launchpad, nlaunches=20, sleep_time=0)
+        self.mc.reset(hard=False)
+        self.mc.configure(wf_creator=wf_creator_basic,
+                          dimensions=self.dims_basic,
+                          predictor="GaussianProcessRegressor",
+                          acq="ei")
+        rapidfire(launchpad, nlaunches=2, sleep_time=0)
+        self.mc.reset(hard=False)
+        self.mc.configure(wf_creator=wf_creator_basic,
+                          dimensions=self.dims_basic,
+                          predictor="GaussianProcessRegressor",
+                          acq="pi")
+        rapidfire(launchpad, nlaunches=2, sleep_time=0)
+        self.mc.reset(hard=False)
+        self.mc.configure(wf_creator=wf_creator_basic,
+                          dimensions=self.dims_basic,
+                          predictor="GaussianProcessRegressor",
+                          acq="lcb")
+        rapidfire(launchpad, nlaunches=2, sleep_time=0)
+
+
 
     def test_parallel(self):
         n_procs = 10

rocketsled/utils.py

@@ -219,6 +219,10 @@ def split_xz(xz, x_dims, x_only=False, z_only=False):
 def get_default_opttask_kwargs():
     """
     Get the default configuration kwargs for OptTask.
+
+    Args:
+        None
+
     Returns:
         conf_dict (dict): The default kwargs for OptTask
 
@@ -248,7 +252,7 @@ def check_dims(dims):
         dims (list): The dimensions of the search space.
 
     Returns:
-        [str]: Types of the dimensions in the search space defined by dims.
+        ([str]): Types of the dimensions in the search space defined by dims.
     """
     dims_types = [list, tuple]
     dim_spec = []