docker-python/tests/test_hyperopt.py at main · SKOGZ22/docker-python · GitHub

Sitelet https://github.com/SKOGZ22/docker-python/blob/main/tests/test_hyperopt.py

13 lines (11 loc) · 316 Bytes

import unittest
from hyperopt import fmin, tpe, hp
class TestHyperopt(unittest.TestCase):
    def test_find_min(self):
        best = fmin(
            fn=lambda x: x ** 2,
            space=hp.uniform('x', -10, 10),
            algo=tpe.suggest,
            max_evals=1,
        self.assertIn('x', best)