def test_call(self): """Test to get task function""" egg = EggHolder(2) assert callable(egg) objectives = egg([1, 2]) assert type(objectives[0]) == dict
def test_call(self): """Test to get task function""" task = EggHolder(2) assert callable(task) objectives = task([1, 2]) assert type(objectives[0]) == dict
def test_creation(self): """Test creation""" branin = EggHolder(max_trials=2, dim=3) assert branin.max_trials == 2 assert branin.configuration == { "EggHolder": { "dim": 3, "max_trials": 2 } }
def test_simple(): """Test a end 2 end exucution of benchmark""" task_num = 2 trial_num = 20 assessments = [AverageResult(task_num), AverageRank(task_num)] tasks = [ RosenBrock(trial_num, dim=3), EggHolder(trial_num, dim=4), CarromTable(trial_num), Branin(trial_num), BirdLike(trial_num), ] benchmark = get_or_create_benchmark( name="bm001", algorithms=algorithms, targets=[{ "assess": assessments, "task": tasks }], ) benchmark.process() assert len(benchmark.studies) == len(assessments) * len(tasks) status = benchmark.status() experiments = benchmark.experiments() assert len(experiments ) == len(algorithms) * task_num * len(assessments) * len(tasks) assert len(status) == len(algorithms) * len(assessments) * len(tasks) figures = benchmark.analysis() assert len(figures) == len(benchmark.studies) assert type(figures[0]) is plotly.graph_objects.Figure benchmark = get_or_create_benchmark(name="bm001") figures = benchmark.analysis() assert len(figures) == len(benchmark.studies) assert type(figures[0]) is plotly.graph_objects.Figure
def test_search_space(self): """Test to get task search space""" egg = EggHolder(2) assert egg.get_search_space() == {"x": "uniform(-512, 512, shape=2)"}