def test_update_gbest_neighborhood(swarm, p, r): """Test if update_gbest_neighborhood gives the expected return values""" topology = VonNeumann() pos, cost = topology.compute_gbest(swarm, p=p, r=r) expected_pos = np.array([9.90438476e-01, 2.50379538e-03, 1.87405987e-05]) expected_cost = 1.0002528364353296 assert cost == pytest.approx(expected_cost) assert pos == pytest.approx(expected_pos)
def test_neighbor_idx(swarm, p, r): """Test if the neighbor_idx attribute is assigned""" topology = VonNeumann() topology.compute_gbest(swarm, p=p, r=r) assert topology.neighbor_idx is not None