def test_search_my_neighbors(self):
glowworm1 = Glowworm(self.landscape_position1, self.gso_parameters)
glowworm2 = Glowworm(self.landscape_position4, self.gso_parameters)
glowworm3 = Glowworm(self.landscape_position5, self.gso_parameters)
glowworm1.luciferin = 3.0
glowworm1.vision_range = 1.0
glowworms = [glowworm1, glowworm2, glowworm3]
glowworm1.search_neighbors(glowworms)
assert 1 == len(glowworm1.neighbors)
assert glowworm1.neighbors[0] == glowworm2
def test_search_my_neighbors_when_distance_equal_range(self):
glowworm1 = Glowworm(self.landscape_position1, self.gso_parameters)
glowworm2 = Glowworm(self.landscape_position4, self.gso_parameters)
glowworm3 = Glowworm(self.landscape_position5, self.gso_parameters)
glowworm4 = Glowworm(self.landscape_position6, self.gso_parameters)
glowworm1.luciferin = 3.0
glowworm1.vision_range = 1.0
glowworms = [glowworm1, glowworm2, glowworm3, glowworm4]
glowworm1.search_neighbors(glowworms)
assert len(glowworm1.neighbors) == 1
assert glowworm1.neighbors[0] == glowworm2