def time_extended_spatial_hash(pa, nbrs, H, approx, num_particles): start_time = time.time() nn = ExtendedSpatialHash(pa,H,approximate = approx) for i in range(num_particles): nn.nearest_neighbours(i,nbrs) nbrs.reset() return time.time() - start_time
def test_exact_extended_spatial_hash_variable_radius(self): print "" for key in self.dataset_variable_radius: x, y, z, h = self.dataset_variable_radius[key] pa = get_particle_array(x=x, y=y, z=z, h=h) qid = randint(0,pa.get_number_of_particles()-1) nn_bf = brute_force_neighbours(pa, pa.x[qid], pa.y[qid], pa.z[qid], pa.h[qid]) exp = ExtendedSpatialHash(pa, 3, approximate = False) nn_sp = UIntArray() exp.nearest_neighbours(qid, nn_sp) print " - " + key assert equal(nn_bf, nn_sp)