ENH check for injection of runhistory as well

automl · Oct 5, 2016 · cbafd7b · cbafd7b
1 parent e581b9d
commit cbafd7b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/test_facade/test_smac_facade.py b/test/test_facade/test_smac_facade.py
@@ -2,8 +2,10 @@
 
 from smac.configspace import ConfigurationSpace
 
+from smac.runhistory.runhistory import RunHistory
 from smac.facade.smac_facade import SMAC
 from smac.scenario.scenario import Scenario
+from smac.stats.stats import Stats
 from smac.tae.execute_func import ExecuteTAFunc
 
 
@@ -13,8 +15,10 @@ def setUp(self):
         self.cs = ConfigurationSpace()
         self.scenario = Scenario({'cs': self.cs, 'run_obj': 'quality'})
 
-    def test_inject_stats_object_to_TAE(self):
+    def test_inject_stats_and_runhistory_object_to_TAE(self):
         ta = ExecuteTAFunc(lambda x: x**2)
         self.assertIsNone(ta.stats)
+        self.assertIsNone(ta.runhistory)
         SMAC(tae_runner=ta, scenario=self.scenario)
-        self.assertIsNotNone(ta.stats)
+        self.assertIsInstance(ta.stats, Stats)
+        self.assertIsInstance(ta.runhistory, RunHistory)