def test_ssd_run(): def square(x): return np.sum(x**2) def hook(optimizer, space, function): return new_function = function.Function(pointer=square) hyperparams = {'c': 2.0, 'decay': 0.3} new_ssd = ssd.SSD(hyperparams=hyperparams) search_space = search.SearchSpace(n_agents=50, n_iterations=350, n_variables=2, lower_bound=[-100, -100], upper_bound=[100, 100]) history = new_ssd.run(search_space, new_function, pre_evaluation=hook) assert len(history.agents) > 0 assert len(history.best_agent) > 0 best_fitness = history.best_agent[-1][1] assert best_fitness <= constants.TEST_EPSILON, 'The algorithm ssd failed to converge.'
def test_ssd_params_setter(): new_ssd = ssd.SSD() try: new_ssd.c = 'a' except: new_ssd.c = 0.5 try: new_ssd.c = -1 except: new_ssd.c = 0.5 assert new_ssd.c == 0.5 try: new_ssd.decay = 'b' except: new_ssd.decay = 0.99 try: new_ssd.decay = -1 except: new_ssd.decay = 0.99 assert new_ssd.decay == 0.99
def test_ssd_params(): params = {'c': 2.0, 'decay': 0.99} new_ssd = ssd.SSD(params=params) assert new_ssd.c == 2.0 assert new_ssd.decay == 0.99
def test_ssd_hyperparams(): hyperparams = {'c': 2.0, 'decay': 0.99} new_ssd = ssd.SSD(hyperparams=hyperparams) assert new_ssd.c == 2.0 assert new_ssd.decay == 0.99
def test_ssd_params(): params = {"c": 2.0, "decay": 0.99} new_ssd = ssd.SSD(params=params) assert new_ssd.c == 2.0 assert new_ssd.decay == 0.99
def test_ssd_update_velocity(): search_space = search.SearchSpace(n_agents=10, n_variables=2, lower_bound=[0, 0], upper_bound=[10, 10]) new_ssd = ssd.SSD() new_ssd.compile(search_space) velocity = new_ssd._update_velocity(0.5, 10, 1) assert velocity[0] != 0
def test_ssd_update_position(): search_space = search.SearchSpace(n_agents=10, n_variables=2, lower_bound=[0, 0], upper_bound=[10, 10]) new_ssd = ssd.SSD() new_ssd.compile(search_space) position = new_ssd._update_position(1, 1) assert position[0][0] != 0
def test_ssd_update(): def square(x): return np.sum(x**2) search_space = search.SearchSpace(n_agents=10, n_variables=2, lower_bound=[0, 0], upper_bound=[10, 10]) new_ssd = ssd.SSD() new_ssd.compile(search_space) new_ssd.update(search_space, square)
def test_ssd_evaluate(): def square(x): return np.sum(x**2) search_space = search.SearchSpace(n_agents=10, n_variables=2, lower_bound=[0, 0], upper_bound=[10, 10]) new_ssd = ssd.SSD() new_ssd.compile(search_space) new_ssd.evaluate(search_space, square) assert search_space.best_agent.fit != constant.FLOAT_MAX
def test_ssd_compile(): search_space = search.SearchSpace(n_agents=10, n_variables=2, lower_bound=[0, 0], upper_bound=[10, 10]) new_ssd = ssd.SSD() new_ssd.compile(search_space) try: new_ssd.local_position = 1 except: new_ssd.local_position = np.array([1]) assert new_ssd.local_position == 1 try: new_ssd.velocity = 1 except: new_ssd.velocity = np.array([1]) assert new_ssd.velocity == 1
def test_ssd_mean_global_solution(): new_ssd = ssd.SSD() mean = new_ssd._mean_global_solution(1, 2, 3) assert mean != 0
def test_ssd_update_position(): new_ssd = ssd.SSD() position = new_ssd._update_position(1, 1) assert position == 2
def test_ssd_update_velocity(): new_ssd = ssd.SSD() velocity = new_ssd._update_velocity(0.5, 10, 25) assert velocity[0] != 0
def test_ssd_build(): new_ssd = ssd.SSD() assert new_ssd.built == True