From 40b01922a10f5d7ddc25d4718390af5d5f35bcae Mon Sep 17 00:00:00 2001 From: KanaiYuma Date: Wed, 6 Mar 2024 10:36:26 +0900 Subject: [PATCH] Add test for waiting_for --- tests/unit_tests/test_nelder_mead_sampler.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/unit_tests/test_nelder_mead_sampler.py b/tests/unit_tests/test_nelder_mead_sampler.py index 1153fb1e2..b837e665d 100644 --- 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()