Add test for waiting_for

aistairc · Mar 6, 2024 · 40b0192 · 40b0192
1 parent 9212e43
commit 40b0192
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/unit_tests/test_nelder_mead_sampler.py b/tests/unit_tests/test_nelder_mead_sampler.py
@@ -27,12 +27,22 @@ def test_waiting_for(self):
         result = yield from self.nm._waiting_for_float()
         self.assertIsNone(result)
 
+        result = yield from self.nm._waiting_for_list(2)
+        self.assertIsNone(result)
+
         # queue is not Empty
         value = 1.0
         self.nm.value_queue.put(value)
         result = yield from self.nm._waiting_for_float()
         self.assertEqual(result, value)
 
+        value1 = 1.0
+        self.nm.value_queue.put(value1)
+        value2 = 2.0
+        self.nm.value_queue.put(value2)
+        result = yield from self.nm._waiting_for_list(2)
+        self.assertEqual(result, [value1, value2])
+
     def test_initialize(self):
         for _ in range(len(self.search_space) + 1):
             xi = self.nm.get_vertex()